Wing Lian 546ad007ec don't use triton for now

2025-01-09 18:57:24 -05:00

..

__init__.py

RL/DPO (#935)

2024-01-04 18:22:55 -05:00

base.py

KD dataset loading and KD with logprobs

2025-01-09 18:57:22 -05:00

kd.py

don't use triton for now

2025-01-09 18:57:24 -05:00

trl.py

feat: add cut_cross_entropy (#2091)

2024-12-03 08:22:22 -05:00