yichenchenchen commited on
Commit
e21fd5e
·
verified ·
1 Parent(s): df8d65c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ import tempfile
6
  import sys
7
  import time
8
  from accelerate.utils import set_seed
9
- from huggingface_hub import snapshot_download, configure_http_backend
10
  import subprocess
11
 
12
  #os.system("pip install ./flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl")
@@ -22,7 +22,7 @@ def ensure_flash_attn():
22
  ensure_flash_attn()
23
  from inferencer import UniPicV2Inferencer
24
 
25
- configure_http_backend("httpx", timeout=60.0)
26
  model_path = snapshot_download(repo_id="Skywork/UniPic2-Metaquery-GRPO-9B")
27
  qwen_vl_path = snapshot_download(repo_id="Qwen/Qwen2.5-VL-7B-Instruct-AWQ")
28
 
 
6
  import sys
7
  import time
8
  from accelerate.utils import set_seed
9
+ from huggingface_hub import snapshot_download
10
  import subprocess
11
 
12
  #os.system("pip install ./flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl")
 
22
  ensure_flash_attn()
23
  from inferencer import UniPicV2Inferencer
24
 
25
+ os.environ["HF_HUB_REQUEST_TIMEOUT"] = "60.0"
26
  model_path = snapshot_download(repo_id="Skywork/UniPic2-Metaquery-GRPO-9B")
27
  qwen_vl_path = snapshot_download(repo_id="Qwen/Qwen2.5-VL-7B-Instruct-AWQ")
28