pllava-7b-demo / app.py
cathyxl
added
f239efc
import sys
from huggingface_hub import snapshot_download
snapshot_download(
'ermu2001/pllava-7b',
local_dir='MODELS/pllava-7b',
repo_type='model',
local_dir_use_symlinks=True,
)
sys.argv.extend([
"--pretrained_model_name_or_path", "MODELS/pllava-7b",
"--num_frames", "16",
"--use_lora",
"--weight_dir", "MODELS/pllava-7b",
"--lora_alpha", "4",
"--conv_mode", "plain",
])
import tasks.eval.demo.pllava_demo