ford442 commited on
Commit
a043550
Β·
1 Parent(s): 71c0675

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -141,7 +141,7 @@ def generate(
141
  global models
142
  pipe = models[model_choice]
143
  seed = int(randomize_seed_fn(seed, randomize_seed))
144
- generator = torch.manual_seed(seed)
145
 
146
  prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)
147
 
 
141
  global models
142
  pipe = models[model_choice]
143
  seed = int(randomize_seed_fn(seed, randomize_seed))
144
+ generator = torch.Generator().manual_seed(seed)
145
 
146
  prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)
147