RealVis_v5.0_BF16_IP

Running on Zero

ford442 commited on Nov 25, 2024

Commit

7afcfd1

1 Parent(s): ebf5804

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -141,7 +141,7 @@ def generate(
     global models
     pipe = models[model_choice]
     seed = int(randomize_seed_fn(seed, randomize_seed))
-    generator = torch.Generator(device=device).manual_seed(seed)
     prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)

     global models
     pipe = models[model_choice]
     seed = int(randomize_seed_fn(seed, randomize_seed))
+    generator = torch.manual_seed(seed)
     prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)