Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -27,14 +27,15 @@ from models.RIFE.IFNet_HDv3 import IFNet
|
|
27 |
|
28 |
device = "cuda"
|
29 |
|
30 |
-
sd_path =
|
31 |
inter_path = "checkpoints/flownet.pkl"
|
32 |
controlnet_dict = {
|
33 |
-
"pose":
|
34 |
-
"depth":
|
35 |
-
"canny":
|
36 |
}
|
37 |
|
|
|
38 |
controlnet_parser_dict = {
|
39 |
"pose": OpenposeDetector,
|
40 |
"depth": MidasDetector,
|
|
|
27 |
|
28 |
device = "cuda"
|
29 |
|
30 |
+
sd_path = "checkpoints/stable-diffusion-v1-5"
|
31 |
inter_path = "checkpoints/flownet.pkl"
|
32 |
controlnet_dict = {
|
33 |
+
"pose": "checkpoints/sd-controlnet-openpose",
|
34 |
+
"depth": "checkpoints/sd-controlnet-depth",
|
35 |
+
"canny": "checkpoints/sd-controlnet-canny",
|
36 |
}
|
37 |
|
38 |
+
|
39 |
controlnet_parser_dict = {
|
40 |
"pose": OpenposeDetector,
|
41 |
"depth": MidasDetector,
|