Dan Saunders 2962a398b7 Lora kernels fix (#2732) ...

* fix lora kernel patching and improve test

* simplification

2025-05-28 10:03:43 -04:00

..

integrations

don't sort multipack sampler (#2657)

2025-05-09 20:28:58 -04:00

kernels

adding pre-commit auto-update GH action and bumping plugin versions (#2428)

2025-03-21 11:02:43 -04:00

multigpu

Rank 0-only logging (#2608)

2025-05-28 14:57:30 +01:00

patched

Lora kernels fix (#2732)

2025-05-28 10:03:43 -04:00

solo

Rank 0-only logging (#2608)

2025-05-28 14:57:30 +01: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_deepseekv3.py

Rank 0-only logging (#2608)

2025-05-28 14:57:30 +01:00

test_dpo.py

Rank 0-only logging (#2608)

2025-05-28 14:57:30 +01:00

test_embeddings_lr.py

Rank 0-only logging (#2608)

2025-05-28 14:57:30 +01:00

test_evaluate.py

swap tinymodels that have safetensors for some ci tests (#2641)

2025-05-07 15:06:07 -04:00

test_falcon.py

Rank 0-only logging (#2608)

2025-05-28 14:57:30 +01:00

test_gemma2.py

Rank 0-only logging (#2608)

2025-05-28 14:57:30 +01:00

test_gemma3_text.py

Rank 0-only logging (#2608)

2025-05-28 14:57:30 +01:00

test_imports.py

adding pre-commit auto-update GH action and bumping plugin versions (#2428)

2025-03-21 11:02:43 -04:00

test_llama_pretrain.py

Rank 0-only logging (#2608)

2025-05-28 14:57:30 +01:00

test_llama_vision.py

Rank 0-only logging (#2608)

2025-05-28 14:57:30 +01:00

test_llama.py

Rank 0-only logging (#2608)

2025-05-28 14:57:30 +01:00

test_load_model.py

models.py -> loaders/ module refactor (#2680)

2025-05-23 15:51:11 -04:00

test_lora_llama.py

Rank 0-only logging (#2608)

2025-05-28 14:57:30 +01:00

test_mamba.py

Rank 0-only logging (#2608)

2025-05-28 14:57:30 +01:00

test_mistral.py

Rank 0-only logging (#2608)

2025-05-28 14:57:30 +01:00

test_mixtral.py

Rank 0-only logging (#2608)

2025-05-28 14:57:30 +01:00

test_optimizers.py

Rank 0-only logging (#2608)

2025-05-28 14:57:30 +01:00

test_packing_loss.py

Rank 0-only logging (#2608)

2025-05-28 14:57:30 +01:00

test_phi.py

Rank 0-only logging (#2608)

2025-05-28 14:57:30 +01:00

test_process_reward_model_smollm2.py

Rank 0-only logging (#2608)

2025-05-28 14:57:30 +01:00

test_qat.py

QAT (#2590)

2025-05-28 12:35:47 +01:00

test_quantization.py

QAT (#2590)

2025-05-28 12:35:47 +01:00

test_qwen.py

Rank 0-only logging (#2608)

2025-05-28 14:57:30 +01:00

test_reward_model_smollm2.py

Rank 0-only logging (#2608)

2025-05-28 14:57:30 +01:00

test_schedulers.py

Rank 0-only logging (#2608)

2025-05-28 14:57:30 +01:00

utils.py

QAT (#2590)

2025-05-28 12:35:47 +01:00