TryingHard commited on
Commit
5cf1f22
·
verified ·
1 Parent(s): dd7bd41

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import subprocess
2
  import os
3
  os.environ['CUDA_VISIBLE_DEVICES']='0'
4
- subprocess.run('pip install flash-attn==2.7.0.post2 --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
5
  subprocess.run('pip install gptqmodel==1.9.0', shell=True)
 
6
  subprocess.run('pip install numpy==1.25.0 torch==2.4.0 transformers==4.46.2 pillow==10.3.0 moviepy==1.0.3', shell=True)
7
  os.environ['CUDA_VISIBLE_DEVICES']='0'
8
 
 
1
  import subprocess
2
  import os
3
  os.environ['CUDA_VISIBLE_DEVICES']='0'
 
4
  subprocess.run('pip install gptqmodel==1.9.0', shell=True)
5
+ subprocess.run('pip install flash-attn==2.7.0.post2 --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
6
  subprocess.run('pip install numpy==1.25.0 torch==2.4.0 transformers==4.46.2 pillow==10.3.0 moviepy==1.0.3', shell=True)
7
  os.environ['CUDA_VISIBLE_DEVICES']='0'
8