fix bug
Browse files
app.py
CHANGED
@@ -96,7 +96,7 @@ def edit(brush_canvas,
|
|
96 |
source_prompt, target_prompt,
|
97 |
inversion_num_steps, denoise_num_steps,
|
98 |
skip_step,
|
99 |
-
|
100 |
re_init,attn_mask,attn_scale_value
|
101 |
):
|
102 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
@@ -133,7 +133,7 @@ def edit(brush_canvas,
|
|
133 |
inversion_num_steps=inversion_num_steps,
|
134 |
denoise_num_steps=denoise_num_steps,
|
135 |
skip_step=skip_step,
|
136 |
-
inversion_guidance=
|
137 |
denoise_guidance=denoise_guidance,
|
138 |
seed=seed,
|
139 |
re_init=re_init,
|
@@ -270,7 +270,6 @@ def create_demo(model_name: str):
|
|
270 |
source_prompt, target_prompt,
|
271 |
denoise_num_steps, denoise_num_steps,
|
272 |
skip_step,
|
273 |
-
1,
|
274 |
denoise_guidance,seed,
|
275 |
re_init,attn_mask,attn_scale_value
|
276 |
],
|
|
|
96 |
source_prompt, target_prompt,
|
97 |
inversion_num_steps, denoise_num_steps,
|
98 |
skip_step,
|
99 |
+
denoise_guidance,seed,
|
100 |
re_init,attn_mask,attn_scale_value
|
101 |
):
|
102 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
133 |
inversion_num_steps=inversion_num_steps,
|
134 |
denoise_num_steps=denoise_num_steps,
|
135 |
skip_step=skip_step,
|
136 |
+
inversion_guidance=1,
|
137 |
denoise_guidance=denoise_guidance,
|
138 |
seed=seed,
|
139 |
re_init=re_init,
|
|
|
270 |
source_prompt, target_prompt,
|
271 |
denoise_num_steps, denoise_num_steps,
|
272 |
skip_step,
|
|
|
273 |
denoise_guidance,seed,
|
274 |
re_init,attn_mask,attn_scale_value
|
275 |
],
|