Spaces:
Runtime error
Runtime error
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
|