Update single_dcu_train.sh

96ac4dfb · dcuai · 78e22504 · 96ac4dfb
Commit 96ac4dfb authored Oct 23, 2024 by dcuai
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

single_dcu_train.sh single_dcu_train.sh +1 -1

No files found.
--- a/single_dcu_train.sh
+++ b/single_dcu_train.sh
 #!/bin/bash

-CUDA_VISIBLE_DEVICES=0 python LLaMA-Factory-main/src/train_bash.py \
+HIP_VISIBLE_DEVICES=0 python LLaMA-Factory-main/src/train_bash.py \
    --stage sft \
    --do_train \
    --model_name_or_path THUDM/chatglm2-6b \