Merge branch 'k100ai_dtk2404_optim'

729150b6 · yinger_z · d8704953 · b0b8940e · 729150b6 · 729150b6
Commit 729150b6 authored Aug 29, 2024 by yinger_z
18 changed files
--- a/ptuning/train_chat.sh
+++ b/ptuning/train_chat.sh
--- a/ptuning/trainer.py
+++ b/ptuning/trainer.py
--- a/ptuning/trainer_lora.py
+++ b/ptuning/trainer_lora.py
--- a/ptuning/trainer_seq2seq.py
+++ b/ptuning/trainer_seq2seq.py
--- a/ptuning/web_demo.py
+++ b/ptuning/web_demo.py
--- a/ptuning/web_demo.sh
+++ b/ptuning/web_demo.sh
--- a/requirements.txt
+++ b/requirements.txt
@@ -11,4 +11,4 @@ datasets
 protobuf
 mpi4py
 peft
-pydantic<2.0
+pydantic==1.10.9
--- a/simple-pretrain/The-Lord-of-the-Rings-1.txt
+++ b/simple-pretrain/The-Lord-of-the-Rings-1.txt
--- a/simple-pretrain/convert.py
+++ b/simple-pretrain/convert.py
--- a/simple-pretrain/modeling_chatglm.py
+++ b/simple-pretrain/modeling_chatglm.py
--- a/simple-pretrain/ptuning/The-Lord-of-the-Rings-1.json
+++ b/simple-pretrain/ptuning/The-Lord-of-the-Rings-1.json
--- a/simple-pretrain/ptuning/deepspeed.json
+++ b/simple-pretrain/ptuning/deepspeed.json
--- a/simple-pretrain/ptuning/ds_pretrain.sh
+++ b/simple-pretrain/ptuning/ds_pretrain.sh
--- a/simple-pretrain/ptuning/main.py
+++ b/simple-pretrain/ptuning/main.py
--- a/utils.py
+++ b/utils.py
--- a/web_demo.py
+++ b/web_demo.py
--- a/web_demo2.py
+++ b/web_demo2.py
--- a/web_demo_old.py
+++ b/web_demo_old.py