fix pipeline

main.py

@@ -19,6 +19,8 @@ pipe = pipe.to(device)
 
 def generate_image(prompt, negative_prompt, seed, guidance_scale, num_inference_steps, model):
 
+    generator = torch.Generator(device=device).manual_seed(seed)
+
     if model == "from-scratch":
         image = pipeline.generate(
             prompt=prompt,
@@ -30,16 +32,12 @@ def generate_image(prompt, negative_prompt, seed, guidance_scale, num_inference_
             width=512,
             height=512,
             generator=generator,
-            models={},
-            seed=seed,
             device=device,
             idle_device="cpu",
+            models=models,
             tokenizer=tokenizer,
         )
-        return image
     else:
-        generator = torch.Generator(device=device).manual_seed(seed)
-
         image = pipe(
             prompt=prompt,
             negative_prompt=negative_prompt,
@@ -50,7 +48,7 @@ def generate_image(prompt, negative_prompt, seed, guidance_scale, num_inference_
             generator=generator,
         ).images[0]
 
-        return image
+    return image
 
 css="""
 #col-container {

sd/pipeline.py

@@ -7,15 +7,15 @@ def generate(
     prompt,
     uncond_prompt=None,
     input_image=None,
-    strength=0.8,
-    cfg_scale=7.5,
+    strength=0.9,
+    cfg_scale=8.0,
     n_inference_steps=50,
     width=512,
     height=512,
     generator=None,
-    models={},
     device=None,
     idle_device=None,
+    models={},
     tokenizer=None,
 ):
     latents_width = width // 8