hiyouga eb70328eec add llava to llamaboard
Former-commit-id: cd3a960f81579b4173ae79aa14d075e94651e8ce
2024-04-26 06:41:35 +08:00

9 lines
273 B
Bash

#!/bin/bash
# add `--visual_inputs True` to load MLLM
CUDA_VISIBLE_DEVICES=0 python ../../src/web_demo.py \
--model_name_or_path meta-llama/Llama-2-7b-hf \
--adapter_name_or_path ../../saves/LLaMA2-7B/lora/sft \
--template default \
--finetuning_type lora