RamAnanth1 commited on
Commit
db0cb98
1 Parent(s): d304db8

Remove strength argument from inpaint pipeline

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -168,7 +168,7 @@ def process_image_inpaint(raw_image,mask, prompt):
168
  torch.manual_seed(SEED)
169
  print(SEED)
170
 
171
- strength = 0.7
172
  guidance_scale = 7.5
173
  num_inference_steps = 100
174
 
@@ -177,8 +177,8 @@ def process_image_inpaint(raw_image,mask, prompt):
177
  mask_image=mask_image,
178
  eta=1,
179
  num_inference_steps=num_inference_steps,
180
- guidance_scale=guidance_scale,
181
- strength=strength
182
  ).images[0]
183
  image_nat = recover_image(image_nat, init_image, mask_image)
184
 
@@ -188,8 +188,8 @@ def process_image_inpaint(raw_image,mask, prompt):
188
  mask_image=mask_image,
189
  eta=1,
190
  num_inference_steps=num_inference_steps,
191
- guidance_scale=guidance_scale,
192
- strength=strength
193
  ).images[0]
194
  image_adv = recover_image(image_adv, init_image, mask_image)
195
 
 
168
  torch.manual_seed(SEED)
169
  print(SEED)
170
 
171
+ #strength = 0.7
172
  guidance_scale = 7.5
173
  num_inference_steps = 100
174
 
 
177
  mask_image=mask_image,
178
  eta=1,
179
  num_inference_steps=num_inference_steps,
180
+ guidance_scale=guidance_scale
181
+ #strength=strength
182
  ).images[0]
183
  image_nat = recover_image(image_nat, init_image, mask_image)
184
 
 
188
  mask_image=mask_image,
189
  eta=1,
190
  num_inference_steps=num_inference_steps,
191
+ guidance_scale=guidance_scale
192
+ #strength=strength
193
  ).images[0]
194
  image_adv = recover_image(image_adv, init_image, mask_image)
195