Update lora_finetune_multi.sh

93652d86 · wanglch · e2e7ab10 · 93652d86
Commit 93652d86 authored Jun 19, 2024 by wanglch
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 4 deletions

lora_finetune_multi.sh lora_finetune_multi.sh +4 -4

No files found.
--- a/lora_finetune_multi.sh
+++ b/lora_finetune_multi.sh
 NPROC_PER_NODE=2 \
-CUDA_VISIBLE_DEVICES=5,6 swift sft \
+HIP_VISIBLE_DEVICES=5,6 swift sft \
-    --model_id_or_path /home/wanglch/projects/GLM-4V/glm-4v-b \
+    --model_id_or_path /home/wanglch/projects/GLM-4V/glm-4v-9b \
    --model_type glm4v-9b-chat \
-    --dataset coco-en-2-mini \
+    --dataset /data/json \
    --output_dir /home/wanglch/projects/saves/glm-4v/train_dtk \
    --num_train_epochs 1 \
    --eval_steps 200 \