Wing Lian c7b6790614 Various fixes for CI, save_only_model for RL, prevent packing multiprocessing deadlocks (#2661) ...

* lean mistral ft tests, remove e2e torch 2.4.1 test

* make sure to pass save_only_model for RL

* more tests to make ci leaner, add cleanup to modal ci

* fix module for import in e2e tests

* use mp spawn to prevent deadlocks with packing

* make sure cleanup shell script is executable when cloned out

2025-05-12 10:51:18 -04:00

..

__init__.py

Various fixes for CI, save_only_model for RL, prevent packing multiprocessing deadlocks (#2661)

2025-05-12 10:51:18 -04:00

cicd.sh

Various fixes for CI, save_only_model for RL, prevent packing multiprocessing deadlocks (#2661)

2025-05-12 10:51:18 -04:00

cleanup.py

Various fixes for CI, save_only_model for RL, prevent packing multiprocessing deadlocks (#2661)

2025-05-12 10:51:18 -04:00

cleanup.sh

Various fixes for CI, save_only_model for RL, prevent packing multiprocessing deadlocks (#2661)

2025-05-12 10:51:18 -04:00

Dockerfile.jinja

Sequence parallelism (#2412)

2025-03-21 12:43:55 -04:00

e2e_tests.py

Various fixes for CI, save_only_model for RL, prevent packing multiprocessing deadlocks (#2661)

2025-05-12 10:51:18 -04:00

multigpu.py

Codecov fixes / improvements (#2549)

2025-04-23 10:33:30 -04:00

multigpu.sh

update trl to 0.17.0 (#2560)

2025-04-27 19:19:53 -04:00

single_gpu.py

Various fixes for CI, save_only_model for RL, prevent packing multiprocessing deadlocks (#2661)

2025-05-12 10:51:18 -04:00