Spaces:
Runtime error
Runtime error
RohitGandikota
commited on
Commit
β’
71c541b
1
Parent(s):
23cbc09
faster inference
Browse files
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=
|
320 |
|
321 |
generator = torch.manual_seed(seed)
|
322 |
-
original_image = self.pipe(prompt, num_images_per_prompt=1, num_inference_steps=
|
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
|