TryingHard commited on
Commit
dd7bd41
·
verified ·
1 Parent(s): 11be32f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ 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==2.1.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
 
 
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