Handle missing xformers CUDA

code/losses.py

@@ -30,7 +30,11 @@ class SDSLoss(nn.Module):
         self.pipe.to(device)
         self.pipe.text_encoder.to(device)
 
-        self.pipe.enable_xformers_memory_efficient_attention()
+        if self.fp16:
+            try:
+                self.pipe.enable_xformers_memory_efficient_attention()
+            except Exception as e:
+                print(f"WARNING: xFormers memory efficient attention could not be enabled: {e}")
         # Use additional VRAM to disable memory-saving features for speed
         if hasattr(self.pipe, "disable_attention_slicing"):
             self.pipe.disable_attention_slicing()