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 = {
|