LPX55 commited on
Commit
ed55c07
·
verified ·
1 Parent(s): 9d038eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from pipelines.hidream_image.pipeline_hidream_image import HiDreamImagePipeline
9
  import subprocess
10
 
11
  print(subprocess.check_output(["nvcc", "--version"]).decode("utf-8"))
12
- subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
13
 
14
  from nf4 import *
15
 
 
9
  import subprocess
10
 
11
  print(subprocess.check_output(["nvcc", "--version"]).decode("utf-8"))
12
+ subprocess.run('python -m pip install flash-attn --no-build-isolation', shell=True)
13
 
14
  from nf4 import *
15