hiyouga e6126244c1 fix possibly OOM error

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