da03 commited on
Commit
f201f64
·
1 Parent(s): 8043e65
Files changed (1) hide show
  1. main.py +3 -0
main.py CHANGED
@@ -36,6 +36,9 @@ 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:
 
36
 
37
  MODEL_NAME = "yuntian-deng/computer-model-ss005-cont-lr2e5-computecanada-2k"
38
 
39
+ MODEL_NAME = "yuntian-deng/computer-model-ss005-cont-lr2e5-computecanada-10k"
40
+
41
+
42
  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}')
43
 
44
  with open('latent_stats.json', 'r') as f: