hiyouga 7ef5821cba fix possibly OOM error ...

Former-commit-id: 0d590dffb41b0e832d9f87d20a23bcd0acd983aa

2023-06-01 23:54:44 +08:00

..

__init__.py

support BLOOM models

2023-05-31 16:54:06 +08:00

common.py

fix possibly OOM error

2023-06-01 23:54:44 +08:00

config.py

support BLOOM models

2023-05-31 16:54:06 +08:00

data_collator.py

support BLOOM models

2023-05-31 16:54:06 +08:00

other.py

support BLOOM models

2023-05-31 16:54:06 +08:00

pairwise.py

support BLOOM models

2023-05-31 16:54:06 +08:00

peft_trainer.py

use fp16 model, add logcallback

2023-05-28 21:30:28 +08:00

ppo.py

support BLOOM models

2023-05-31 16:54:06 +08:00

seq2seq.py

support BLOOM models

2023-05-31 16:54:06 +08:00