Wing Lian 5e9fa33f3d reduce test concurrency to avoid HF rate limiting, test suite parity (#2128) ...

* reduce test concurrency to avoid HF rate limiting, test suite parity

* make val_set_size smaller to speed up e2e tests

* more retries for pytest fixture downloads

* val_set_size was too small

* move retry_on_request_exceptions to data utils and add retry strategy

* pre-download ultrafeedback as a test fixture

* refactor download retry into it's own fn

* don't import from data utils

* use retry mechanism now for fixtures

2024-12-06 10:20:20 -05:00

..

integrations

feat: add cut_cross_entropy (#2091)

2024-12-03 08:22:22 -05:00

multigpu

upgrade bnb 0.45.0 and peft 0.14.0 (#2126)

2024-12-06 09:08:55 -05:00

patched

reduce test concurrency to avoid HF rate limiting, test suite parity (#2128)

2024-12-06 10:20:20 -05:00

__init__.py

missing dunder-init

2023-11-06 18:33:01 -05:00

.gitignore

Support Sample packing for phi arch (#586)

2023-09-15 15:46:54 -04:00

test_dpo.py

Add weighted optimisation support for trl DPO trainer integration (#2016)

2024-11-08 11:29:11 -05:00

test_embeddings_lr.py

replace tensorboard checks with helper function (#2120) [skip ci]

2024-12-03 21:06:20 -05:00

test_falcon.py

Falcon embeddings (#1149) [skip docker]

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

test_imports.py

Bump deepspeed 20240727 (#1790)

2024-07-27 10:24:11 -04:00

test_llama_pretrain.py

fixes to accelerator so that iterable pretraining datasets work (#1759)

2024-07-17 10:58:38 -04:00

test_llama_vision.py

fix(vlm): handle legacy conversation data format and check image in data (#2018) [skip ci]

2024-12-03 00:01:31 -05:00

test_llama.py

Update get_unpad_data patching for multipack (#2013)

2024-11-15 20:35:50 -05:00

test_load_model.py

Refactor func load_model to class ModelLoader (#1909)

2024-10-25 09:06:56 -04:00

test_lora_llama.py

fix(vlm): handle legacy conversation data format and check image in data (#2018) [skip ci]

2024-12-03 00:01:31 -05:00

test_mamba.py

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

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

test_mistral.py

use temp_dir kwarg instead

2023-11-06 18:33:01 -05:00

test_mixtral.py

fix broken linting (#1541)

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

test_optimizers.py

fix(vlm): handle legacy conversation data format and check image in data (#2018) [skip ci]

2024-12-03 00:01:31 -05:00

test_packing_loss.py

replace tensorboard checks with helper function (#2120) [skip ci]

2024-12-03 21:06:20 -05:00

test_phi.py

re-enable phi for tests in modal ci (#1373)

2024-05-29 15:41:46 -04:00

test_qwen.py

remove the bos token from dpo outputs (#1733) [skip ci]

2024-11-15 19:09:20 -05:00

test_relora_llama.py

replace tensorboard checks with helper function (#2120) [skip ci]

2024-12-03 21:06:20 -05:00

test_reward_model_llama.py

Reward model (#1879)

2024-10-13 15:11:13 -04:00

utils.py

replace tensorboard checks with helper function (#2120) [skip ci]

2024-12-03 21:06:20 -05:00