RealVis_v5.0_BF16_IP

Running on Zero

ford442 commited on Dec 9, 2024

Commit

e74e0f4

1 Parent(s): 99a67da

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -170,7 +170,7 @@ def generate_60(
     global models
     pipe = models[model_choice]
     seed = int(randomize_seed_fn(seed, randomize_seed))
-    generator = torch.Generator(device='cpu').manual_seed(seed)
     prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)
     options = {
         "prompt": [prompt] * num_images,
@@ -220,7 +220,7 @@ def generate_90(
     global models
     pipe = models[model_choice]
     seed = int(randomize_seed_fn(seed, randomize_seed))
-    generator = torch.Generator(device='cpu').manual_seed(seed)
     prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)
     options = {
         "prompt": [prompt] * num_images,

     global models
     pipe = models[model_choice]
     seed = int(randomize_seed_fn(seed, randomize_seed))
+    generator = torch.Generator(device='cuda').manual_seed(seed)
     prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)
     options = {
         "prompt": [prompt] * num_images,
     global models
     pipe = models[model_choice]
     seed = int(randomize_seed_fn(seed, randomize_seed))
+    generator = torch.Generator(device='cuda').manual_seed(seed)
     prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)
     options = {
         "prompt": [prompt] * num_images,