Update app.py

app.py

@@ -63,6 +63,8 @@ BATCH_SIZE = int(os.getenv("BATCH_SIZE", "1"))
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
+generator = torch.Generator(device=device)
+
 style_list = [
     {
         "name": "3840 x 2160",
@@ -165,7 +167,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)
+    generator.manual_seed(seed)
 
     prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)