Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -26,12 +26,13 @@ from models.RIFE.IFNet_HDv3 import IFNet
|
|
26 |
|
27 |
|
28 |
device = "cuda"
|
29 |
-
|
|
|
30 |
inter_path = "checkpoints/flownet.pkl"
|
31 |
controlnet_dict = {
|
32 |
-
"pose": "lllyasviel/sd-controlnet-pose",
|
33 |
-
"depth": "lllyasviel/sd-controlnet-depth",
|
34 |
-
"canny": "lllyasviel/sd-controlnet-canny",
|
35 |
}
|
36 |
|
37 |
controlnet_parser_dict = {
|
|
|
26 |
|
27 |
|
28 |
device = "cuda"
|
29 |
+
|
30 |
+
sd_path = hf_hub_download("runwayml/stable-diffusion-v1-5", cache_dir="checkpoints")
|
31 |
inter_path = "checkpoints/flownet.pkl"
|
32 |
controlnet_dict = {
|
33 |
+
"pose": hf_hub_download("lllyasviel/sd-controlnet-pose", cache_dir="checkpoints") ,
|
34 |
+
"depth": hf_hub_download("lllyasviel/sd-controlnet-depth", cache_dir="checkpoints"),
|
35 |
+
"canny": hf_hub_download("lllyasviel/sd-controlnet-canny", cache_dir="checkpoints")
|
36 |
}
|
37 |
|
38 |
controlnet_parser_dict = {
|