multimodalart HF Staff commited on
Commit
80784a2
·
verified ·
1 Parent(s): fb0638e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import spaces
2
  import subprocess
3
  # Installing flash_attn
4
- subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
5
 
6
  import torch
7
  from diffusers import AutoencoderKLWan, WanImageToVideoPipeline, UniPCMultistepScheduler
 
1
  import spaces
2
  import subprocess
3
  # Installing flash_attn
4
+ subprocess.run('pip install flash-attn==2.7.4.post1 --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
5
 
6
  import torch
7
  from diffusers import AutoencoderKLWan, WanImageToVideoPipeline, UniPCMultistepScheduler