Skip to content

GitLab

Commit 9d70560e authored by wanglch's avatar wanglch
Browse files

Replace finetune_lora_single_gpu.sh

parent fecd1c9f
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
finetune/finetune_lora_single_gpu.sh
View file @ 9d70560e
...@@ -6,13 +6,14 @@ DIR=`pwd` ...@@ -6,13 +6,14 @@ DIR=`pwd`
MODEL="Qwen/Qwen-VL-Chat" #"Qwen/Qwen-VL-Chat"/"Qwen/Qwen-VL" # Set the path if you do not want to load from huggingface directly MODEL="Qwen/Qwen-VL-Chat" #"Qwen/Qwen-VL-Chat"/"Qwen/Qwen-VL" # Set the path if you do not want to load from huggingface directly
# ATTENTION: specify the path to your training data, which should be a json file consisting of a list of conversations. # ATTENTION: specify the path to your training data, which should be a json file consisting of a list of conversations.
# See the section for finetuning in README for more information. # See the section for finetuning in README for more information.
DATA="path_to_data" DATA="./data.json"
export CUDA_VISIBLE_DEVICES=0 export CUDA_VISIBLE_DEVICES=0
python finetune.py \ python finetune.py \
--model_name_or_path $MODEL \ --model_name_or_path $MODEL \
--data_path $DATA \ --data_path $DATA \
--bf16 True \
--fix_vit True \ --fix_vit True \
--output_dir output_qwen \ --output_dir output_qwen \
--num_train_epochs 5 \ --num_train_epochs 5 \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment