github
/
ChatGLM2-6B
mirror of https://github.com/THUDM/ChatGLM2-6B
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Update use accelerate cmd

pull/672/head
wanghh2000 2023-12-28 10:59:38 +08:00
parent 771ad3ac93
commit fe24caa129
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ptuning/finetune-p-tuning-v2.py
View File

@ -19,9 +19,9 @@ Fine-tuning the library models for sequence to sequence for P-Tuning v2
# You can also adapt this script on your own sequence to sequence task. Pointers for this are left as comments.
# CUDA_VISIBLE_DEVICES=-1 python finetune-p-tuning-v2.py
# accelerate launch --cpu --num_machines=1 --num_processes=1 --num_cpu_threads_per_process=1 finetune-p-tuning-v2.py
# accelerate launch --cpu --num_machines=1 --num_processes=4 --num_cpu_threads_per_process=1 finetune-p-tuning-v2.py
# accelerate launch --num_machines=1 --num_processes=1 --gpu_ids=1 finetune-p-tuning-v2.py
# import logging
import os