da03 commited on
Commit
7323319
·
1 Parent(s): f146b8e
Files changed (1) hide show
  1. worker.py +2 -2
worker.py CHANGED
@@ -53,7 +53,7 @@ class GPUWorker:
53
  self.NUM_SAMPLING_STEPS = 32
54
  self.USE_RNN = False
55
 
56
- self.MODEL_NAME = "yuntian-deng/computer-model-s-newnewd-freezernn-origunet-nospatial-online-x0-joint-onlineonly-222222k7-06k"
57
 
58
  # Initialize model
59
  self._initialize_model()
@@ -810,4 +810,4 @@ if __name__ == "__main__":
810
  logger.error(f"❌ Failed to start worker: {e}")
811
  import traceback
812
  logger.error(f"🔍 Full traceback: {traceback.format_exc()}")
813
- raise
 
53
  self.NUM_SAMPLING_STEPS = 32
54
  self.USE_RNN = False
55
 
56
+ self.MODEL_NAME = "yuntian-deng/computer-model-s-newnewd-freezernn-origunet-nospatial-online-x0-joint-onlineonly-222222k72-108k"
57
 
58
  # Initialize model
59
  self._initialize_model()
 
810
  logger.error(f"❌ Failed to start worker: {e}")
811
  import traceback
812
  logger.error(f"🔍 Full traceback: {traceback.format_exc()}")
813
+ raise