Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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.
|
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'
|