"tests/nn/vscode:/vscode.git/clone" did not exist on "bc1e60e09c2437c284284e09c11ec9e6767a998b"
Commit 9d70560e authored by wanglch's avatar wanglch
Browse files

Replace finetune_lora_single_gpu.sh

parent fecd1c9f
......@@ -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
# 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.
DATA="path_to_data"
DATA="./data.json"
export CUDA_VISIBLE_DEVICES=0
python finetune.py \
--model_name_or_path $MODEL \
--data_path $DATA \
--bf16 True \
--fix_vit True \
--output_dir output_qwen \
--num_train_epochs 5 \
......@@ -33,4 +34,4 @@ python finetune.py \
--model_max_length 2048 \
--lazy_preprocess True \
--gradient_checkpointing \
--use_lora
--use_lora
\ No newline at end of file
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