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,
|