olfp commited on
Commit
b7b5e4d
1 Parent(s): 7d77915

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -15,12 +15,16 @@ 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 snapshot_download
19
 
20
 
21
  snapshot_download(repo_id="xichenhku/AnyDoor_models", local_dir="./AnyDoor_models")
22
  snapshot_download(repo_id="xichenhku/mask_refine", local_dir="./mask_refine")
23
 
 
 
24
  cv2.setNumThreads(0)
25
  cv2.ocl.setUseOpenCL(False)
26
 
 
15
  from cldm.ddim_hacked import DDIMSampler
16
  from omegaconf import OmegaConf
17
  from cldm.hack import disable_verbosity, enable_sliced_attention
18
+ from transformers import AutoModel
19
+
20
  from huggingface_hub import snapshot_download
21
 
22
 
23
  snapshot_download(repo_id="xichenhku/AnyDoor_models", local_dir="./AnyDoor_models")
24
  snapshot_download(repo_id="xichenhku/mask_refine", local_dir="./mask_refine")
25
 
26
+ model = AutoModel.from_pretrained("private/model", token=access_token)
27
+
28
  cv2.setNumThreads(0)
29
  cv2.ocl.setUseOpenCL(False)
30