Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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"
|