|
model_dir=${1:-"MODELS/pllava-7b"} |
|
weight_dir=${2:-"${model_dir}"} |
|
num_frames=16 |
|
lora_alpha=4 |
|
|
|
echo Running DEMO from model_dir: ${model_dir} |
|
echo Running DEMO from weights_dir: ${weight_dir} |
|
echo Running DEMO On Devices: ${CUDA_VISIBLE_DEVICES} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
CUDA_VISIBLE_DEVICES=${CUDA_VISIBLE_DEVICES:-"0,1"} python -m tasks.eval.demo.pllava_demo \ |
|
--pretrained_model_name_or_path ${model_dir} \ |
|
--num_frames ${num_frames} \ |
|
--use_lora \ |
|
--weight_dir ${weight_dir} \ |
|
--lora_alpha ${lora_alpha} \ |
|
--conv_mode plain \ |
|
--use_multi_gpus |
|
|
|
|
|
|