RohitGandikota commited on
Commit
71c541b
β€’
1 Parent(s): 23cbc09

faster inference

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -316,10 +316,10 @@ class Demo:
316
 
317
 
318
  generator = torch.manual_seed(seed)
319
- edited_image = self.pipe(prompt, num_images_per_prompt=1, num_inference_steps=50, generator=generator, network=network, start_noise=int(start_noise), scale=float(scale), unet=unet).images[0]
320
 
321
  generator = torch.manual_seed(seed)
322
- original_image = self.pipe(prompt, num_images_per_prompt=1, num_inference_steps=50, generator=generator, network=network, start_noise=start_noise, scale=0, unet=unet).images[0]
323
 
324
  del unet, network
325
  unet = None
 
316
 
317
 
318
  generator = torch.manual_seed(seed)
319
+ edited_image = self.pipe(prompt, num_images_per_prompt=1, num_inference_steps=20, generator=generator, network=network, start_noise=int(start_noise), scale=float(scale), unet=unet).images[0]
320
 
321
  generator = torch.manual_seed(seed)
322
+ original_image = self.pipe(prompt, num_images_per_prompt=1, num_inference_steps=20, generator=generator, network=network, start_noise=start_noise, scale=0, unet=unet).images[0]
323
 
324
  del unet, network
325
  unet = None