Wing Lian d87df2c776 prepare plugins needs to happen so registration can occur to build the plugin args (#2119) ...

* prepare plugins needs to happen so registration can occur to build the plugin args

use yaml.dump

include dataset and more assertions

* attempt to manually register plugins rather than use fn

* fix fixture

* remove fixture

* move cli test to patched dir

* fix cce validation

2024-12-03 15:06:09 -05:00

..

__init__.py

attempt to also run e2e tests that needs gpus (#1070)

2024-01-09 21:23:23 -05:00

test_4d_multipack_llama.py

Fix: modelloader handling of model_kwargs load_in*bit (#1999)

2024-10-30 14:41:34 -04:00

test_cli_integrations.py

prepare plugins needs to happen so registration can occur to build the plugin args (#2119)

2024-12-03 15:06:09 -05:00

test_fa_xentropy.py

various tests fixes for flakey tests (#2110)

2024-12-02 17:28:58 -05:00

test_falcon_samplepack.py

Falcon embeddings (#1149) [skip docker]

2024-01-22 21:01:42 -05:00

test_fused_llama.py

make mlflow optional (#1317)

2024-02-26 11:41:33 -05:00

test_llama_s2_attention.py

support for gemma2 w sample packing (#1718)

2024-06-29 01:38:55 -04:00

test_lora_llama_multipack.py

optim e2e tests to run a bit faster (#2069) [skip ci]

2024-11-18 12:35:31 -05:00

test_mistral_samplepack.py

relora: magnitude pruning of the optimizer (#1245)

2024-02-06 00:35:30 -05:00

test_mixtral_samplepack.py

fix broken linting (#1541)

2024-04-19 01:03:04 -04:00

test_model_patches.py

swaps to use newer sample packing for mistral (#1773)

2024-07-23 01:41:11 -04:00

test_phi_multipack.py

Phi2 multipack (#1173)

2024-01-23 12:54:36 -05:00

test_resume.py

make the eval size smaller for the resume test (#2111) [skip ci]

2024-12-02 18:32:29 -05:00

test_unsloth_integration.py

Fix: Gradient Accumulation issue (#1980)

2024-10-25 11:28:23 -04:00

test_unsloth_qlora.py

various tests fixes for flakey tests (#2110)

2024-12-02 17:28:58 -05:00