fffiloni commited on
Commit
3541d3a
·
1 Parent(s): a02fbe2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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 = "checkpoints/stable-diffusion-v1-5"
32
  inter_path = "checkpoints/flownet.pkl"
33
  controlnet_dict = {
34
- "pose": "checkpoints/sd-controlnet-openpose",
35
- "depth": "checkpoints/sd-controlnet-depth",
36
- "canny": "checkpoints/sd-controlnet-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