Add xFormers library form attention calculate

app.py

@@ -185,6 +185,14 @@ pipe = TryonPipeline.from_pretrained(
 pipe.unet_encoder = UNet_Encoder
 print("✓ TryonPipeline initialized successfully")
 
+# xFormers 메모리 효율적 어텐션 활성화 (메모리 20-30% 감소, 속도 10-20% 향상)
+print("\n[Optimization] Enabling xFormers memory efficient attention...")
+try:
+    pipe.enable_xformers_memory_efficient_attention()
+    print("✓ xFormers memory efficient attention enabled")
+except Exception as e:
+    print(f"⚠ xFormers not available, using default attention: {e}")
+
 print("\n" + "=" * 60)
 print("All models loaded successfully!")
 print("=" * 60 + "\n")

requirements.txt

@@ -14,6 +14,7 @@ matplotlib==3.7.4
 tqdm==4.64.1
 config==0.5.1
 einops==0.7.0
+xformers>=0.0.22
 onnxruntime==1.16.2
 basicsr
 av