olfp commited on
Commit
1bc25c0
1 Parent(s): ba6d3e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -15,10 +15,10 @@ from cldm.model import create_model, load_state_dict
15
  from cldm.ddim_hacked import DDIMSampler
16
  from omegaconf import OmegaConf
17
  from cldm.hack import disable_verbosity, enable_sliced_attention
18
- from huggingface_hub import hf_hub_download
19
 
20
- hf_hub_download(repo_id="onlineformapro/anydoor-models-ofp", filename="general_v0.1.ckpt")
21
- hf_hub_download(repo_id="onlineformapro/anydoor-refine-ofp", filename="coarse_mask_refine.pth")
22
 
23
  cv2.setNumThreads(0)
24
  cv2.ocl.setUseOpenCL(False)
 
15
  from cldm.ddim_hacked import DDIMSampler
16
  from omegaconf import OmegaConf
17
  from cldm.hack import disable_verbosity, enable_sliced_attention
18
+ from huggingface_hub import snapshot_download
19
 
20
+ snapshot_download(repo_id="onlineformapro/anydoor-models-ofp", local_dir="./anydoor-models-ofp")
21
+ snapshot_download(repo_id="onlineformapro/anydoor-refine-ofp", local_dir="./anydoor-refine-ofp")
22
 
23
  cv2.setNumThreads(0)
24
  cv2.ocl.setUseOpenCL(False)