Update app.py
Browse files
app.py
CHANGED
|
@@ -142,7 +142,7 @@ def predict(dict, prompt="", negative_prompt = default_negative_prompt, guidance
|
|
| 142 |
# mask_resized = mask_resized.to(torch.float16)
|
| 143 |
masked_image = torch.cat([control_latents, mask_resized], dim=1)
|
| 144 |
|
| 145 |
-
generator = torch.Generator(device='cuda').manual_seed(
|
| 146 |
|
| 147 |
output = pipe(prompt = prompt,
|
| 148 |
width=width,
|
|
|
|
| 142 |
# mask_resized = mask_resized.to(torch.float16)
|
| 143 |
masked_image = torch.cat([control_latents, mask_resized], dim=1)
|
| 144 |
|
| 145 |
+
generator = torch.Generator(device='cuda').manual_seed(int(seed))
|
| 146 |
|
| 147 |
output = pipe(prompt = prompt,
|
| 148 |
width=width,
|