da03 commited on
Commit
34b3fb7
·
1 Parent(s): f5a1d14
Files changed (1) hide show
  1. main.py +2 -0
main.py CHANGED
@@ -34,6 +34,8 @@ MODEL_NAME = "yuntian-deng/computer-model-ss005-cont-lr2e5-384k"
34
 
35
  MODEL_NAME = "yuntian-deng/computer-model-noss-forsure"
36
 
 
 
37
  print (f'setting: DEBUG_MODE: {DEBUG_MODE}, DEBUG_MODE_2: {DEBUG_MODE_2}, NUM_MAX_FRAMES: {NUM_MAX_FRAMES}, NUM_SAMPLING_STEPS: {NUM_SAMPLING_STEPS}, MODEL_NAME: {MODEL_NAME}')
38
 
39
  with open('latent_stats.json', 'r') as f:
 
34
 
35
  MODEL_NAME = "yuntian-deng/computer-model-noss-forsure"
36
 
37
+ MODEL_NAME = "yuntian-deng/computer-model-ss005-cont-lr2e5-computecanada-2k"
38
+
39
  print (f'setting: DEBUG_MODE: {DEBUG_MODE}, DEBUG_MODE_2: {DEBUG_MODE_2}, NUM_MAX_FRAMES: {NUM_MAX_FRAMES}, NUM_SAMPLING_STEPS: {NUM_SAMPLING_STEPS}, MODEL_NAME: {MODEL_NAME}')
40
 
41
  with open('latent_stats.json', 'r') as f: