Wing Lian 23f029a89c lint and additional train metric checks for kd

2025-02-26 03:19:42 -05:00

..

__init__.py

add e2e smoke tests for llama liger integration (#1884)

2024-09-01 19:29:37 -04:00

test_cut_cross_entropy.py

updating to fused (#2293)

2025-01-30 11:45:56 -05:00

test_kd.py

lint and additional train metric checks for kd

2025-02-26 03:19:42 -05:00

test_kl_loss.py

lint and additional train metric checks for kd

2025-02-26 03:19:42 -05:00

test_liger.py

KD Trainer w logprobs (#2303)

2025-01-31 20:18:52 -05:00

test_logsumexp.py

fix gradients

2025-02-25 23:34:27 -05:00