xilluill commited on
Commit
6587d24
·
1 Parent(s): cdf7291
Files changed (1) hide show
  1. app.py +2 -3
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
- inversion_guidance, denoise_guidance,seed,
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=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
  ],