Wing Lian c0c53eb62f various tests fixes for flakey tests (#2110) ...

* add mhenrichsen/alpaca_2k_test with revision dataset download fixture for flaky tests

* log slowest tests

* pin pynvml==11.5.3

* fix load local hub path

* optimize for speed w smaller models and val_set_size

* replace pynvml

* make the resume from checkpoint e2e faster

* make tests smaller

2024-12-04 12:33:29 -05:00

..

cicd.sh

various tests fixes for flakey tests (#2110)

2024-12-04 12:33:29 -05:00

Dockerfile.jinja

add e2e tests for Unsloth qlora and test the builds (#2093)

2024-12-04 12:33:29 -05:00

multigpu.py

change deprecated modal Stub to App (#2038)

2024-11-11 15:10:34 -05:00

multigpu.sh

gradient accumulation tests, embeddings w pad_token fix, smaller models (#2059)

2024-11-14 12:59:00 -05:00

tests.py

change deprecated modal Stub to App (#2038)

2024-11-11 15:10:34 -05:00