Update app.py

app.py

@@ -91,7 +91,7 @@ pipe.fuse_lora()
 pipe = pipe.to(device)
 # pipe.enable_xformers_memory_efficient_attention()
 
-generator = torch.Generator(device='cuda').manual_seed(123456)
+# generator = torch.Generator(device='cuda').manual_seed(123456)
 
 vae = pipe.vae
 
@@ -105,7 +105,7 @@ def read_content(file_path: str) -> str:
 
     return content
 
-def predict(dict, prompt="", negative_prompt = default_negative_prompt, guidance_scale=1.2, steps=12, strength=1.0):
+def predict(dict, prompt="", negative_prompt = default_negative_prompt, guidance_scale=1.2, steps=12, strength=1.0, seed=123456):
     if negative_prompt == "":
         negative_prompt = None
 
@@ -142,7 +142,8 @@ def predict(dict, prompt="", negative_prompt = default_negative_prompt, guidance
     # mask_resized = mask_resized.to(torch.float16)
     masked_image = torch.cat([control_latents, mask_resized], dim=1)
 
-
+    generator = torch.Generator(device='cuda').manual_seed(123456)
+    
     output = pipe(prompt = prompt,
                   width=width,
                   height=height,
@@ -218,9 +219,10 @@ with image_blocks as demo:
                     
                     with gr.Accordion(label="Advanced Settings", open=False):
                         with gr.Row(equal_height=True):
-                            guidance_scale = gr.Number(value=1.2, minimum=1.0, maximum=2, step=0.1, label="guidance_scale")
-                            steps = gr.Number(value=12, minimum=8, maximum=30, step=1, label="steps")
-                            strength = gr.Number(value=1, minimum=0.01, maximum=1.0, step=0.01, label="strength")
+                            guidance_scale = gr.Number(value=1.2, minimum=0.8, maximum=2.5, step=0.1, label="guidance_scale")
+                            steps = gr.Number(value=12, minimum=6, maximum=20, step=1, label="steps")
+                            # strength = gr.Number(value=1, minimum=0.01, maximum=1.0, step=0.01, label="strength")
+                            seed = gr.Number(value=123456, minimum=0, maximum=999999, step=1, label="seed")
                             negative_prompt = gr.Textbox(label="negative_prompt", value=default_negative_prompt, placeholder=default_negative_prompt, info="what you don't want to see in the image")
 
                         
@@ -229,7 +231,7 @@ with image_blocks as demo:
 
             
 
-    btn.click(fn=predict, inputs=[image, prompt, negative_prompt, guidance_scale, steps, strength], outputs=[image_out], api_name='run')
+    btn.click(fn=predict, inputs=[image, prompt, negative_prompt, guidance_scale, steps, strength, seed], outputs=[image_out], api_name='run')
     prompt.submit(fn=predict, inputs=[image, prompt, negative_prompt, guidance_scale, steps, strength], outputs=[image_out])
 
     gr.HTML(