Wing Lian 7e410ab480 more fixes to flex for fsdp2

2025-04-06 14:24:50 -04:00

..

solo

Flex Attention + Packing with BlockMask support (#2363)

2025-04-05 18:02:57 -04:00

__init__.py

Attempt to run multigpu in PR CI for now to ensure it works (#1815) [skip ci]

2024-08-09 11:50:13 -04:00

test_eval.py

Updates for trl 0.16.0 - mostly for GRPO (#2437) [skip ci]

2025-03-31 15:47:11 -04:00

test_gemma3.py

gemma3 packing fixes (#2449)

2025-03-31 17:15:23 -04:00

test_llama.py

more fixes to flex for fsdp2

2025-04-06 14:24:50 -04:00

test_qwen2.py

Update dependencies and show slow tests in CI (#2492)

2025-04-05 17:41:31 -04:00

test_ray.py

Updates for trl 0.16.0 - mostly for GRPO (#2437) [skip ci]

2025-03-31 15:47:11 -04:00