Update gradio_demo.py

gradio_demo.py

@@ -107,15 +107,15 @@ def enable_efficient_attention():
             print(f"Xformers error: {e}")
             print("Falling back to sliced attention")
             # Use sliced attention for RTX 2070
-            unet.set_attention_slice_size(4)
-            vae.set_attention_slice_size(4)
+            #unet.set_attention_slice_size(4)
+            #vae.set_attention_slice_size(4)
             unet.set_attn_processor(AttnProcessor2_0())
             vae.set_attn_processor(AttnProcessor2_0())
     else:
         # Fallback for when xformers is not available
         print("Using sliced attention")
-        unet.set_attention_slice_size(4)
-        vae.set_attention_slice_size(4)
+        #unet.set_attention_slice_size(4)
+        #vae.set_attention_slice_size(4)
         unet.set_attn_processor(AttnProcessor2_0())
         vae.set_attn_processor(AttnProcessor2_0())
 
@@ -652,13 +652,13 @@ quick_subjects = [[x] for x in quick_subjects]
 
 
 class BGSource(Enum):
-    UPLOAD = "Use Background Image"
-    UPLOAD_FLIP = "Use Flipped Background Image"
     LEFT = "Left Light"
     RIGHT = "Right Light"
     TOP = "Top Light"
     BOTTOM = "Bottom Light"
     GREY = "Ambient"
+    UPLOAD = "Use Background Image"
+    UPLOAD_FLIP = "Use Flipped Background Image"
 
 # Add save function
 def save_images(images, prefix="relight"):