Dan Saunders 208fb7b8e7 basic torchao fp8 mixed precision training (#2926) ...

* debug

* debug

* debug

* revert unneeded change

* add accelerator config to base trainer builder

* add back accumulated_cache_size_limit setting

* lint

* accelerator constructor patch for single-GPU torch fp8

* lint

* re-using existing fp8 code

* lint

* remove accelerate patch now fix in latest release

* fix

* docs

* add fp8 + fsdp2 example

* remove unused config

* update config

* smoke tests

* add validator

* add 2.7.0 guard for fsdp2

* fix

* add config descriptions

* add FSDP doc link

* nit

* set force_recompute_fp8_weight_in_bwd with enable_fsdp_float8_all_gather

* better cfg for smoke tests

* add test for accelerate patching

* update fp8 validator

2025-07-22 16:27:47 -04:00

..

cli

add mocks for loading datasets in cli train tests (#2497) [skip ci]

2025-04-11 09:51:59 -04:00

core

limit num_proc when saving datasets to disk (#2948) [skip ci]

2025-07-21 11:39:38 -04:00

e2e

basic torchao fp8 mixed precision training (#2926)

2025-07-22 16:27:47 -04:00

fixtures

Respect sequence_len in config for type: llama2_chat (#926)

2023-12-12 09:39:22 -08:00

integrations

remove deprecated wandb env var (#2751)

2025-06-03 14:04:15 -07:00

monkeypatch

basic torchao fp8 mixed precision training (#2926)

2025-07-22 16:27:47 -04:00

patched

release v0.11.0 (#2875)

2025-07-09 09:22:35 -04:00

prompt_strategies

feat: add call method to mistral tokenizer wrapper (#2898)

2025-07-14 22:33:35 -04:00

utils/schemas/validation

make pad_to_sequence_len default to the same value as sample_packing (#2941) [skip ci]

2025-07-21 11:40:56 -04:00

__init__.py

fix: minor patches for multimodal (#2441)

2025-03-31 13:40:12 +07:00

conftest.py

refactor dupes from merge/rebase (#2919) [skip ci]

2025-07-14 10:05:26 -04:00

constants.py

Add Exact Deduplication Feature to Preprocessing Pipeline (#2072)

2024-12-02 08:47:10 -05:00

hf_offline_utils.py

fix tokenizer overrides w gemma3 (#2488)

2025-04-05 01:25:44 -04:00

test_chunked_xentropy.py

chunked cross entropy loss (#2625)

2025-06-23 23:08:46 -04:00

test_data.py

fix: minor patches for multimodal (#2441)

2025-03-31 13:40:12 +07:00

test_datasets.py

limit num_proc when saving datasets to disk (#2948) [skip ci]

2025-07-21 11:39:38 -04:00

test_dict.py

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

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

test_exact_deduplication.py

limit num_proc when saving datasets to disk (#2948) [skip ci]

2025-07-21 11:39:38 -04:00

test_expand_mask.py

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

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

test_freeze.py

Train parameters exclusively in specific ranges (#1390)

2024-03-14 11:05:42 -04:00

test_loaders.py

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

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

test_lora.py

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

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

test_normalize_config.py

FSDP2 fix validation and add tests (#2910)

2025-07-14 09:25:44 -04:00

test_packed_batch_sampler.py

allow for different sequence_len for evaluations (#2836) [skip ci]

2025-06-27 11:02:51 -04:00

test_packed_dataset.py

limit num_proc when saving datasets to disk (#2948) [skip ci]

2025-07-21 11:39:38 -04:00

test_packed_pretraining.py

fix streaming packing test (#2454)

2025-03-29 08:30:06 -04:00

test_perplexity.py

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

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

test_prompt_tokenizers.py

remove deprecated wandb env var (#2751)

2025-06-03 14:04:15 -07:00

test_prompters.py

fix: prompt phi (#1845) [skip ci]

2024-08-22 11:46:57 -04:00

test_schedulers.py

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

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

test_tokenizers.py

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

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

test_train.py

refactor dupes from merge/rebase (#2919) [skip ci]

2025-07-14 10:05:26 -04:00

test_validation_dataset.py

release v0.11.0 (#2875)

2025-07-09 09:22:35 -04:00