RealVis_v5.0_BF16_IP_B

Running on Zero

ford442 commited on Nov 25, 2024

Commit

9f0ce3e

1 Parent(s): 2774ee0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -142,7 +142,7 @@ def generate(
     global models
     pipe = models[model_choice]
     seed = int(randomize_seed_fn(seed, randomize_seed))
-    generator = torch.Generator().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.Generator(device=device).manual_seed(seed)
     prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)