yonishafir commited on
Commit
4c6dcb6
1 Parent(s): 0b58f45

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(123456)
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,