Update app.py

app.py

@@ -162,7 +162,7 @@ def generate(
     width: int = 768,
     height: int = 768,
     guidance_scale: float = 5,
-    num_inference_steps: int = 325,
+    num_inference_steps: int = 300,
     randomize_seed: bool = False,
     use_resolution_binning: bool = True, 
     num_images: int = 1,  
@@ -171,7 +171,7 @@ def generate(
     global models
     pipe = models[model_choice]
     seed = int(randomize_seed_fn(seed, randomize_seed))
-    generator = torch.Generator(device='cpu').manual_seed(seed)
+    generator = torch.Generator(device='cuda').manual_seed(seed)
 
     prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)
 
@@ -366,7 +366,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
                 minimum=10,
                 maximum=1000,
                 step=10,
-                value=325,
+                value=300,
             )
 
     gr.Examples(