nami0342 commited on
Commit
50da29c
Β·
1 Parent(s): e591f25

Remove xFormers, Add it later after torch and etc version aligning

Browse files
Files changed (2) hide show
  1. app.py +8 -7
  2. 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
- # xFormers λ©”λͺ¨λ¦¬ 효율적 μ–΄ν…μ…˜ ν™œμ„±ν™” (λ©”λͺ¨λ¦¬ 20-30% κ°μ†Œ, 속도 10-20% ν–₯상)
189
- print("\n[Optimization] Enabling xFormers memory efficient attention...")
190
- try:
191
- pipe.enable_xformers_memory_efficient_attention()
192
- print("βœ“ xFormers memory efficient attention enabled")
193
- except Exception as e:
194
- print(f"⚠ xFormers not available, using default attention: {e}")
 
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