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