RohitGandikota commited on
Commit
118ce23
Β·
1 Parent(s): 130c313

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -327,10 +327,10 @@ class Demo:
327
 
328
 
329
  generator = torch.manual_seed(seed)
330
- edited_image = self.pipe(prompt, num_images_per_prompt=1, num_inference_steps=1, generator=generator, network=network, start_noise=int(start_noise), scale=float(scale), unet=unet, guidance_scale=1).images[0]
331
 
332
  generator = torch.manual_seed(seed)
333
- original_image = self.pipe(prompt, num_images_per_prompt=1, num_inference_steps=1, generator=generator, network=network, start_noise=start_noise, scale=0, unet=unet, guidance_scale=1).images[0]
334
 
335
  del unet, network
336
  unet = None
 
327
 
328
 
329
  generator = torch.manual_seed(seed)
330
+ edited_image = self.pipe(prompt, num_images_per_prompt=1, num_inference_steps=3, generator=generator, network=network, start_noise=int(start_noise), scale=float(scale), unet=unet, guidance_scale=1).images[0]
331
 
332
  generator = torch.manual_seed(seed)
333
+ original_image = self.pipe(prompt, num_images_per_prompt=1, num_inference_steps=3, generator=generator, network=network, start_noise=start_noise, scale=0, unet=unet, guidance_scale=1).images[0]
334
 
335
  del unet, network
336
  unet = None