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):
|