Spaces:
Runtime error
Runtime error
RamAnanth1
commited on
Commit
•
daf30b7
1
Parent(s):
68e52a7
Update app.py
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ pose_checkpoint = "models/control_sd15_openpose.pth"
|
|
22 |
canny_model = create_model('./models/cldm_v15.yaml')
|
23 |
canny_model.load_state_dict(load_state_dict(cached_download(
|
24 |
hf_hub_url(REPO_ID, canny_checkpoint)
|
25 |
-
), location='
|
26 |
ddim_sampler = DDIMSampler(canny_model)
|
27 |
|
28 |
|
|
|
22 |
canny_model = create_model('./models/cldm_v15.yaml')
|
23 |
canny_model.load_state_dict(load_state_dict(cached_download(
|
24 |
hf_hub_url(REPO_ID, canny_checkpoint)
|
25 |
+
), location='cuda'))
|
26 |
ddim_sampler = DDIMSampler(canny_model)
|
27 |
|
28 |
|