Update prepare.sh

This commit is contained in:
hiyouga 2024-05-02 17:16:02 +08:00
parent 9433c8c215
commit 39e964a97a
1 changed files with 1 additions and 0 deletions

View File

@ -1,4 +1,5 @@
#!/bin/bash #!/bin/bash
# use `--tokenized_path` in training script to load data
CUDA_VISIBLE_DEVICES= python ../../src/train_bash.py \ CUDA_VISIBLE_DEVICES= python ../../src/train_bash.py \
--stage sft \ --stage sft \