Hugging Face's logo

Spaces:
yuntian-group
/
neural-os
Runtime error

App Files Community
yuntian-deng commited on
Commit
3540358
·
verified ·
1 Parent(s): ac4c384

Update worker.py

Browse files
Files changed (1) hide show
  1. worker.py +1 -0
worker.py CHANGED
@@ -54,6 +54,7 @@ class GPUWorker:
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()
 
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
+ self.MODEL_NAME = "yuntian-deng/computer-model-s-newnewd-freezernn-origunet-nospatial-online-x0-joint-onlineonly-222222k722-130k"
58
 
59
  # Initialize model
60
  self._initialize_model()