hiyouga e6126244c1 fix possibly OOM error

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

..

utils

fix possibly OOM error

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

__init__.py

Initial commit

2023-05-28 18:09:04 +08:00

cli_demo.py

fix bug at inference

2023-05-31 18:11:53 +08:00

export_model.py

support BLOOM models

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

train_ppo.py

support BLOOM models

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

train_pt.py

support BLOOM models

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

train_rm.py

support BLOOM models

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

train_sft.py

support BLOOM models

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

web_demo.py

fix bug at inference

2023-05-31 18:11:53 +08:00