Hugging Face's logo

Spaces:
Mountchicken
/
Rex-Omni
Running on Zero

App Files Community
1
Mountchicken commited on
Commit
a171f59
·
verified ·
1 Parent(s): 490ba3d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import json
6
  import os
7
 
8
 
9
- os.system("pip install torch==2.4.0 torchvision==0.18.0 --index-url https://download.pytorch.org/whl/cu124")
10
 
11
  import subprocess
12
  subprocess.run('pip install flash-attn==2.7.4.post1 --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
6
  import os
7
 
8
 
9
+ # os.system("pip install torch==2.4.0 torchvision==0.18.0 --index-url https://download.pytorch.org/whl/cu124")
10
 
11
  import subprocess
12
  subprocess.run('pip install flash-attn==2.7.4.post1 --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)