Nicous commited on
Commit
68d5ac4
·
verified ·
1 Parent(s): b3e9440

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -36,6 +36,8 @@ from egogpt.constants import (
36
  from egogpt.conversation import conv_templates, SeparatorStyle
37
  import subprocess
38
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
 
39
 
40
  # pretrained = "/mnt/sfs-common/jkyang/EgoGPT/checkpoints/EgoGPT-llavaov-7b-EgoIT-109k-release"
41
  # pretrained = "/mnt/sfs-common/jkyang/EgoGPT/checkpoints/EgoGPT-llavaov-7b-EgoIT-EgoLife-Demo"
 
36
  from egogpt.conversation import conv_templates, SeparatorStyle
37
  import subprocess
38
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
39
+ subprocess.run('huggingface-cli download lmms-lab/llava-onevision-qwen2-7b-ov --cache-dir ./llava-onevision-qwen2-7b-ov')
40
+ subprocess.run('huggingface-cli download EgoLife-v1/EgoGPT/large-v3.pt --cache-dir ./large-v3.pt')
41
 
42
  # pretrained = "/mnt/sfs-common/jkyang/EgoGPT/checkpoints/EgoGPT-llavaov-7b-EgoIT-109k-release"
43
  # pretrained = "/mnt/sfs-common/jkyang/EgoGPT/checkpoints/EgoGPT-llavaov-7b-EgoIT-EgoLife-Demo"