Spaces:
Paused
Paused
Update app.py
Browse files
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 |
|