Mountchicken commited on
Commit
17a3917
·
verified ·
1 Parent(s): 6af436d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,13 +18,13 @@ 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 torchvision")
21
- os.system("pip install numpy==1.26.4")
22
  import subprocess
23
  subprocess.run('pip install flash-attn==2.7.4.post1 --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
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
-
28
  import re
29
  from typing import Any, Dict, List
30
 
 
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 torchvision")
21
+
22
  import subprocess
23
  subprocess.run('pip install flash-attn==2.7.4.post1 --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
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