Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,25 +6,13 @@ import json
|
|
| 6 |
import os
|
| 7 |
|
| 8 |
|
| 9 |
-
os.system("pip install matplotlib==3.10.6")
|
| 10 |
-
|
| 11 |
-
os.system("pip install Pillow==11.3.0")
|
| 12 |
-
os.system("pip install qwen_vl_utils==0.0.14")
|
| 13 |
-
os.system("pip install transformers==4.51.3")
|
| 14 |
-
# os.system("pip install vllm==0.8.2")
|
| 15 |
-
os.system("pip install accelerate==1.10.1")
|
| 16 |
-
|
| 17 |
-
os.system("pip install gradio==4.44.1")
|
| 18 |
-
os.system("pip install gradio_image_prompter==0.1.0")
|
| 19 |
-
# os.system("pip install pydantic==2.10.6")
|
| 20 |
os.system("pip install torch==2.4.0 torchvision==0.18.0 --index-url https://download.pytorch.org/whl/cu124")
|
| 21 |
|
| 22 |
-
|
| 23 |
-
|
|
|
|
| 24 |
import sys
|
| 25 |
import threading
|
| 26 |
-
# os.system("pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu124")
|
| 27 |
-
os.system("pip install --no-cache-dir --force-reinstall 'numpy==1.26.4'")
|
| 28 |
import re
|
| 29 |
from typing import Any, Dict, List
|
| 30 |
|
|
@@ -951,4 +939,4 @@ if __name__ == "__main__":
|
|
| 951 |
server_port=args.server_port,
|
| 952 |
share=True,
|
| 953 |
debug=True,
|
| 954 |
-
)
|
|
|
|
| 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)
|
| 13 |
+
|
| 14 |
import sys
|
| 15 |
import threading
|
|
|
|
|
|
|
| 16 |
import re
|
| 17 |
from typing import Any, Dict, List
|
| 18 |
|
|
|
|
| 939 |
server_port=args.server_port,
|
| 940 |
share=True,
|
| 941 |
debug=True,
|
| 942 |
+
)
|