yichenchenchen commited on
Commit
968c544
·
verified ·
1 Parent(s): b93495a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -9,6 +9,8 @@ from inferencer_metaquery import UniPicV2Inferencer
9
  from accelerate.utils import set_seed
10
  from huggingface_hub import snapshot_download
11
 
 
 
12
  model_path = snapshot_download(repo_id="Skywork/UniPic2-Metaquery-9B")
13
  qwen_vl_path = snapshot_download(repo_id="Qwen/Qwen2.5-VL-7B-Instruct-AWQ")
14
 
 
9
  from accelerate.utils import set_seed
10
  from huggingface_hub import snapshot_download
11
 
12
+ os.system("pip install ./flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl")
13
+
14
  model_path = snapshot_download(repo_id="Skywork/UniPic2-Metaquery-9B")
15
  qwen_vl_path = snapshot_download(repo_id="Qwen/Qwen2.5-VL-7B-Instruct-AWQ")
16