Spaces:
Runtime error
Runtime error
Commit
Β·
118ce23
1
Parent(s):
130c313
Update app.py
Browse files
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=
|
331 |
|
332 |
generator = torch.manual_seed(seed)
|
333 |
-
original_image = self.pipe(prompt, num_images_per_prompt=1, num_inference_steps=
|
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
|