Wing Lian dd26cc3c0f add helper to verify the correct model output file exists (#2245) ...

* add helper to verify the correct model output file exists

* more checks using helper

* chore: lint

* fix import and relora model check

* workaround for trl trainer saves

* remove stray print

2025-01-13 10:43:29 -05:00

..

integrations

add helper to verify the correct model output file exists (#2245)

2025-01-13 10:43:29 -05:00

multigpu

need to update deepspeed version in extras too (#2161) [skip ci]

2024-12-09 14:01:44 -05:00

patched

add helper to verify the correct model output file exists (#2245)

2025-01-13 10:43:29 -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 helper to verify the correct model output file exists (#2245)

2025-01-13 10:43:29 -05:00

test_embeddings_lr.py

add helper to verify the correct model output file exists (#2245)

2025-01-13 10:43:29 -05:00

test_falcon.py

add helper to verify the correct model output file exists (#2245)

2025-01-13 10:43:29 -05:00

test_imports.py

Bump deepspeed 20240727 (#1790)

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

test_llama_pretrain.py

add helper to verify the correct model output file exists (#2245)

2025-01-13 10:43:29 -05:00

test_llama_vision.py

add helper to verify the correct model output file exists (#2245)

2025-01-13 10:43:29 -05:00

test_llama.py

add helper to verify the correct model output file exists (#2245)

2025-01-13 10:43:29 -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

add helper to verify the correct model output file exists (#2245)

2025-01-13 10:43:29 -05:00

test_mamba.py

add helper to verify the correct model output file exists (#2245)

2025-01-13 10:43:29 -05:00

test_mistral.py

add helper to verify the correct model output file exists (#2245)

2025-01-13 10:43:29 -05:00

test_mixtral.py

add helper to verify the correct model output file exists (#2245)

2025-01-13 10:43:29 -05:00

test_optimizers.py

add helper to verify the correct model output file exists (#2245)

2025-01-13 10:43:29 -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

add helper to verify the correct model output file exists (#2245)

2025-01-13 10:43:29 -05: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

add helper to verify the correct model output file exists (#2245)

2025-01-13 10:43:29 -05:00

test_reward_model_llama.py

add helper to verify the correct model output file exists (#2245)

2025-01-13 10:43:29 -05:00

utils.py

add helper to verify the correct model output file exists (#2245)

2025-01-13 10:43:29 -05:00