Spaces:
Runtime error
Runtime error
da03
commited on
Commit
·
7323319
1
Parent(s):
f146b8e
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-
|
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
|