Spaces:

Robert001
/

UniControl-Demo

Runtime error

Robert001 commited on Jul 25, 2023

Commit

9b83e5e

•

1 Parent(s): b524d3d

first commit

Files changed (1) hide show

annotator/midas/api.py CHANGED Viewed

@@ -31,7 +31,7 @@ ISL_PATHS = {
 }
 # remote_model_path = "https://huggingface.co/lllyasviel/ControlNet/resolve/main/annotator/ckpts/dpt_hybrid-midas-501f0c75.pt"
-remote_model_path = "https://huggingface.co/Salesforce/UniControl/resolve/main/annotator/ckpts/dpt_large_384.pt"  #"https://huggingface.co/Salesforce/UniControl/blob/main/annotator/ckpts/dpt_large_384.pt"
 def disabled_train(self, mode=True):
     """Overwrite model.train with this function to make sure train/eval mode

 }
 # remote_model_path = "https://huggingface.co/lllyasviel/ControlNet/resolve/main/annotator/ckpts/dpt_hybrid-midas-501f0c75.pt"
+remote_model_path = "https://storage.googleapis.com/sfr-unicontrol-data-research/annotator/ckpts/dpt_large_384.pt"  #"https://huggingface.co/Salesforce/UniControl/blob/main/annotator/ckpts/dpt_large_384.pt"
 def disabled_train(self, mode=True):
     """Overwrite model.train with this function to make sure train/eval mode