Robert001 commited on
Commit
b524d3d
1 Parent(s): 53e9657

first commit

Browse files
Files changed (1) hide show
  1. annotator/midas/api.py +1 -1
annotator/midas/api.py CHANGED
@@ -31,7 +31,7 @@ ISL_PATHS = {
31
  }
32
 
33
  # remote_model_path = "https://huggingface.co/lllyasviel/ControlNet/resolve/main/annotator/ckpts/dpt_hybrid-midas-501f0c75.pt"
34
- remote_model_path = https://huggingface.co/Robert001/UniControl-Model/resolve/main/unicontrol_v1.1.ckpt #"https://huggingface.co/Salesforce/UniControl/blob/main/annotator/ckpts/dpt_large_384.pt"
35
 
36
  def disabled_train(self, mode=True):
37
  """Overwrite model.train with this function to make sure train/eval mode
 
31
  }
32
 
33
  # remote_model_path = "https://huggingface.co/lllyasviel/ControlNet/resolve/main/annotator/ckpts/dpt_hybrid-midas-501f0c75.pt"
34
+ 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"
35
 
36
  def disabled_train(self, mode=True):
37
  """Overwrite model.train with this function to make sure train/eval mode