Update app.py

app.py

@@ -35,13 +35,13 @@ precision = 'fp16'
 # Check for different hardware architectures
 if torch.cuda.is_available():
     device = "cuda"
-    # Check for xformers
-    try:
-        import xformers
+    # # Check for xformers
+    # try:
+    #     import xformers
 
-        enable_xformers = True
-    except ImportError:
-        enable_xformers = False
+    #     enable_xformers = True
+    # except ImportError:
+    #     enable_xformers = False
 elif torch.backends.mps.is_available():
     device = "mps"
 else:
@@ -71,8 +71,9 @@ pipe = pipe.to(device)
 if device == "cuda":
     pipe = pipe.to(device)
     if enable_xformers:
-        pipe.enable_xformers_memory_efficient_attention()
-        print("xformers optimization enabled")
+        # pipe.enable_xformers_memory_efficient_attention()
+        # print("xformers optimization enabled")
+        pipe.enable_attention_slicing()
 elif device == "mps":
     pipe = pipe.to(device)
     pipe.enable_attention_slicing()
@@ -187,11 +188,11 @@ with block:
             with gr.Accordion("Advanced options", open=False):
                 num_samples = gr.Slider(label="Images", minimum=1, maximum=1, value=1, step=1, visible=False)
                 # num_samples = 1
-                image_resolution = gr.Slider(label="Image Resolution", minimum=256, maximum=768, value=768, step=64)
+                image_resolution = gr.Slider(label="Image Resolution", minimum=256, maximum=512, value=512, step=64)
                 strength = gr.Slider(label="Control Strength", minimum=0.0, maximum=2.0, value=1.0, step=0.01)
                 # guess_mode = gr.Checkbox(label='Guess Mode', value=False)
                 ddim_steps = gr.Slider(label="Steps", minimum=1, maximum=20, value=20, step=1)
-                scale = gr.Slider(label="Guidance Scale", minimum=0.1, maximum=30.0, value=4.0, step=0.1)
+                scale = gr.Slider(label="Guidance Scale", minimum=0.1, maximum=30.0, value=7.0, step=0.1)
                 threshold = gr.Slider(label="Segmentation Threshold", minimum=0.1, maximum=0.9, value=0.5, step=0.05)
                 seed = gr.Slider(label="Seed", minimum=-1, maximum=2147483647, value=-1, step=1)
                 eta = gr.Number(label="eta (DDIM)", value=0.0)