Spaces:
Runtime error
Runtime error
RamAnanth1
commited on
Commit
•
9b5dbca
1
Parent(s):
1287f22
Update app.py
Browse files
app.py
CHANGED
@@ -27,7 +27,7 @@ ddim_sampler = DDIMSampler(canny_model)
|
|
27 |
|
28 |
|
29 |
scribble_model = create_model('./models/cldm_v15.yaml')
|
30 |
-
|
31 |
hf_hub_url(REPO_ID, scribble_checkpoint)
|
32 |
), location='cpu'))
|
33 |
scribble_model = canny_model.cuda()
|
|
|
27 |
|
28 |
|
29 |
scribble_model = create_model('./models/cldm_v15.yaml')
|
30 |
+
scribble_model.load_state_dict(load_state_dict(cached_download(
|
31 |
hf_hub_url(REPO_ID, scribble_checkpoint)
|
32 |
), location='cpu'))
|
33 |
scribble_model = canny_model.cuda()
|