Spaces:
Build error
Build error
Clean code
Browse files- .idea/workspace.xml +1 -1
- app.py +0 -2
.idea/workspace.xml
CHANGED
@@ -46,7 +46,7 @@
|
|
46 |
<option name="number" value="Default" />
|
47 |
<option name="presentableId" value="Default" />
|
48 |
<updated>1677222707113</updated>
|
49 |
-
<workItem from="1677222708286" duration="
|
50 |
</task>
|
51 |
<servers />
|
52 |
</component>
|
|
|
46 |
<option name="number" value="Default" />
|
47 |
<option name="presentableId" value="Default" />
|
48 |
<updated>1677222707113</updated>
|
49 |
+
<workItem from="1677222708286" duration="4501000" />
|
50 |
</task>
|
51 |
<servers />
|
52 |
</component>
|
app.py
CHANGED
@@ -17,7 +17,6 @@ canny_checkpoint = "models/control_sd15_canny.pth"
|
|
17 |
scribble_checkpoint = "models/control_sd15_scribble.pth"
|
18 |
pose_checkpoint = "models/control_sd15_openpose.pth"
|
19 |
|
20 |
-
|
21 |
canny_model = create_model('./models/cldm_v15.yaml').cpu()
|
22 |
canny_model.load_state_dict(load_state_dict(cached_download(
|
23 |
hf_hub_url(REPO_ID, canny_checkpoint)
|
@@ -38,7 +37,6 @@ scribble_model.load_state_dict(load_state_dict(cached_download(
|
|
38 |
), location='cpu'))
|
39 |
scribble_model = scribble_model.cuda()
|
40 |
ddim_sampler_scribble = DDIMSampler(scribble_model)
|
41 |
-
|
42 |
save_memory = False
|
43 |
|
44 |
def process(input_image, prompt, input_control, num_samples, image_resolution, ddim_steps, scale, seed, eta, low_threshold, high_threshold):
|
|
|
17 |
scribble_checkpoint = "models/control_sd15_scribble.pth"
|
18 |
pose_checkpoint = "models/control_sd15_openpose.pth"
|
19 |
|
|
|
20 |
canny_model = create_model('./models/cldm_v15.yaml').cpu()
|
21 |
canny_model.load_state_dict(load_state_dict(cached_download(
|
22 |
hf_hub_url(REPO_ID, canny_checkpoint)
|
|
|
37 |
), location='cpu'))
|
38 |
scribble_model = scribble_model.cuda()
|
39 |
ddim_sampler_scribble = DDIMSampler(scribble_model)
|
|
|
40 |
save_memory = False
|
41 |
|
42 |
def process(input_image, prompt, input_control, num_samples, image_resolution, ddim_steps, scale, seed, eta, low_threshold, high_threshold):
|