Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -28,12 +28,12 @@ hf_token = os.environ.get('HF_TOKEN')
|
|
| 28 |
device = "cuda"
|
| 29 |
|
| 30 |
#snapshot_download("runwayml/stable-diffusion-v1-5", cache_dir="checkpoints")
|
| 31 |
-
sd_path = "
|
| 32 |
inter_path = "checkpoints/flownet.pkl"
|
| 33 |
controlnet_dict = {
|
| 34 |
-
"pose": "
|
| 35 |
-
"depth": "
|
| 36 |
-
"canny": "
|
| 37 |
}
|
| 38 |
|
| 39 |
|
|
|
|
| 28 |
device = "cuda"
|
| 29 |
|
| 30 |
#snapshot_download("runwayml/stable-diffusion-v1-5", cache_dir="checkpoints")
|
| 31 |
+
sd_path = "runwayml/stable-diffusion-v1-5"
|
| 32 |
inter_path = "checkpoints/flownet.pkl"
|
| 33 |
controlnet_dict = {
|
| 34 |
+
"pose": "lllyasviel/sd-controlnet-openpose",
|
| 35 |
+
"depth": "lllyasviel/sd-controlnet-depth",
|
| 36 |
+
"canny": "lllyasviel/sd-controlnet-canny",
|
| 37 |
}
|
| 38 |
|
| 39 |
|