Spaces:
Running
on
Zero
Running
on
Zero
yonishafir
commited on
Commit
•
4c6dcb6
1
Parent(s):
0b58f45
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,
|