Spaces:
Sleeping
Sleeping
Remove xFormers, Add it later after torch and etc version aligning
Browse files- app.py +8 -7
- requirements.txt +0 -1
app.py
CHANGED
|
@@ -185,13 +185,14 @@ pipe = TryonPipeline.from_pretrained(
|
|
| 185 |
pipe.unet_encoder = UNet_Encoder
|
| 186 |
print("β TryonPipeline initialized successfully")
|
| 187 |
|
| 188 |
-
#
|
| 189 |
-
|
| 190 |
-
|
| 191 |
-
|
| 192 |
-
|
| 193 |
-
|
| 194 |
-
|
|
|
|
| 195 |
|
| 196 |
print("\n" + "=" * 60)
|
| 197 |
print("All models loaded successfully!")
|
|
|
|
| 185 |
pipe.unet_encoder = UNet_Encoder
|
| 186 |
print("β TryonPipeline initialized successfully")
|
| 187 |
|
| 188 |
+
# torch, diffusers λ± λ²μ μ 리 ν μ μ© κ°λ₯.
|
| 189 |
+
# # xFormers λ©λͺ¨λ¦¬ ν¨μ¨μ μ΄ν
μ
νμ±ν (λ©λͺ¨λ¦¬ 20-30% κ°μ, μλ 10-20% ν₯μ)
|
| 190 |
+
# print("\n[Optimization] Enabling xFormers memory efficient attention...")
|
| 191 |
+
# try:
|
| 192 |
+
# pipe.enable_xformers_memory_efficient_attention()
|
| 193 |
+
# print("β xFormers memory efficient attention enabled")
|
| 194 |
+
# except Exception as e:
|
| 195 |
+
# print(f"β xFormers not available, using default attention: {e}")
|
| 196 |
|
| 197 |
print("\n" + "=" * 60)
|
| 198 |
print("All models loaded successfully!")
|
requirements.txt
CHANGED
|
@@ -14,7 +14,6 @@ matplotlib==3.7.4
|
|
| 14 |
tqdm==4.64.1
|
| 15 |
config==0.5.1
|
| 16 |
einops==0.7.0
|
| 17 |
-
xformers==0.0.22
|
| 18 |
onnxruntime==1.16.2
|
| 19 |
basicsr
|
| 20 |
av
|
|
|
|
| 14 |
tqdm==4.64.1
|
| 15 |
config==0.5.1
|
| 16 |
einops==0.7.0
|
|
|
|
| 17 |
onnxruntime==1.16.2
|
| 18 |
basicsr
|
| 19 |
av
|