Mountchicken commited on
Commit
63abb76
·
verified ·
1 Parent(s): 3945b56

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -16
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
- os.environ["FLASH_ATTENTION_SKIP_CUDA_BUILD"] = "TRUE"
23
- os.system("pip install flash-attn==2.7.4.post1 --no-build-isolation")
 
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
+ )