gokaygokay commited on
Commit
079ebc3
·
verified ·
1 Parent(s): 34903d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ from diffusers import UNet2DConditionModel, AutoencoderKL
15
  from diffusers import EulerDiscreteScheduler
16
 
17
  import subprocess
18
- subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
19
 
20
  # Initialize models
21
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
15
  from diffusers import EulerDiscreteScheduler
16
 
17
  import subprocess
18
+ subprocess.run('pip install flash-attn')
19
 
20
  # Initialize models
21
  device = "cuda" if torch.cuda.is_available() else "cpu"