Spaces:
Runtime error
Runtime error
RamAnanth1
commited on
Commit
•
570118a
1
Parent(s):
02583a6
Update app.py
Browse files
app.py
CHANGED
@@ -37,7 +37,7 @@ ddim_sampler_pose = DDIMSampler(pose_model)
|
|
37 |
scribble_model = create_model('./models/cldm_v15.yaml').cpu()
|
38 |
scribble_model.load_state_dict(load_state_dict(cached_download(
|
39 |
hf_hub_url(REPO_ID, scribble_checkpoint)
|
40 |
-
), location='
|
41 |
scribble_model = canny_model.cuda()
|
42 |
ddim_sampler_scribble = DDIMSampler(scribble_model)
|
43 |
|
|
|
37 |
scribble_model = create_model('./models/cldm_v15.yaml').cpu()
|
38 |
scribble_model.load_state_dict(load_state_dict(cached_download(
|
39 |
hf_hub_url(REPO_ID, scribble_checkpoint)
|
40 |
+
), location='cpu'))
|
41 |
scribble_model = canny_model.cuda()
|
42 |
ddim_sampler_scribble = DDIMSampler(scribble_model)
|
43 |
|