Spaces:
Runtime error
Runtime error
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 |
|