da03 commited on
Commit
552b730
·
1 Parent(s): f201f64
Files changed (1) hide show
  1. main.py +3 -0
main.py CHANGED
@@ -38,6 +38,9 @@ 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
 
 
38
 
39
  MODEL_NAME = "yuntian-deng/computer-model-ss005-cont-lr2e5-computecanada-10k"
40
 
41
+ MODEL_NAME = "yuntian-deng/computer-model-ss005-cont-lr2e5-computecanada-54k"
42
+
43
+
44
 
45
  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}')
46