Wing Lian c011405117 fix gradients

2025-02-25 23:34:27 -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

feat: add config for optional parameters in a chat message (#2260)

2025-02-18 09:59:27 +07:00

test_kl_loss.py

fix gradients

2025-02-25 23:34:27 -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