Update app.py

app.py

@@ -35,11 +35,19 @@ pipe.set_adapters(["better"], adapter_weights=[1.0])
 pipe.fuse_lora(adapter_name=["better"], lora_scale=1.0)
 pipe.unload_lora_weights()
 
-# Corrected: Access 'transformer' instead of 'unet'
+# Correctly set memory format
 pipe.transformer.to(memory_format=torch.channels_last)
 pipe.vae.to(memory_format=torch.channels_last)
 
-pipe.enable_xformers_memory_efficient_attention()
+# Conditionally enable xformers only for the transformer
+if hasattr(pipe, "transformer") and torch.cuda.is_available():
+    try:
+        pipe.transformer.enable_xformers_memory_efficient_attention()
+    except Exception as e:
+        print(
+            "Warning: Could not enable xformers for the transformer due to the following error:"
+        )
+        print(e)
 
 torch.cuda.empty_cache()