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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
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'
 
1
  import subprocess
2
  import os
3
  os.environ['CUDA_VISIBLE_DEVICES']='0'
4
+ subprocess.run('pip install gptqmodel==2.1.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'