ColossalAI/applications/ColossalChat/coati/models
YeAnbang 0b2d55c4ab Support overall loss, update KTO logging 2024-08-02 06:51:38 +00:00
..
__init__.py [Chat] Fix lora (#5946) 2024-07-31 14:10:17 +08:00
base.py add kto 2024-07-18 07:54:11 +00:00
critic.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2024-06-07 07:01:31 +00:00
generation.py [ColossalChat] Update RLHF V2 (#5286) 2024-03-29 14:12:29 +08:00
lora.py [Chat] Fix lora (#5946) 2024-07-31 14:10:17 +08:00
loss.py Support overall loss, update KTO logging 2024-08-02 06:51:38 +00:00
reward_model.py [pre-commit.ci] pre-commit autoupdate (#5572) 2024-07-01 17:16:41 +08:00
utils.py fix dataloader 2024-06-24 05:10:44 +00:00