Sourab Mangrulkar a761d6e9a0 Refactoring Trainer, adds save_only_model arg and simplifying FSDP integration (#27652) ...

* add code changes

1. Refactor FSDP
2. Add `--save_only_model` option: When checkpointing, whether to only save the model, or also the optimizer, scheduler & rng state.
3. Bump up the minimum `accelerate` version to `0.21.0`

* quality

* fix quality?

* Revert "fix quality?"

This reverts commit 149330a6abc078827be274db84c8a2d26a76eba1.

* fix fsdp doc strings

* fix quality

* Update src/transformers/training_args.py

Co-authored-by: Zach Mueller <muellerzr@gmail.com>

* please fix the quality issue 😅

* Apply suggestions from code review

Co-authored-by: Benjamin Bossan <BenjaminBossan@users.noreply.github.com>

* address comment

* simplify conditional check as per the comment

* update documentation

---------

Co-authored-by: Zach Mueller <muellerzr@gmail.com>
Co-authored-by: Benjamin Bossan <BenjaminBossan@users.noreply.github.com>

2023-11-24 11:40:52 +05:30

..

internal

Generate: add missing logits processors docs (#25653)

2023-08-25 11:56:17 +01:00

main_classes

Refactoring Trainer, adds save_only_model arg and simplifying FSDP integration (#27652)

2023-11-24 11:40:52 +05:30

model_doc

[DPT, Dinov2] Add resources (#27655)

2023-11-23 17:44:08 +00:00

tasks

Extended semantic segmentation to image segmentation (#27039)

2023-11-23 15:58:21 +00:00

_config.py

[Styling] stylify using ruff (#27144)

2023-11-16 17:43:19 +01:00

_redirects.yml

Extended semantic segmentation to image segmentation (#27039)

2023-11-23 15:58:21 +00:00

_toctree.yml

Extended semantic segmentation to image segmentation (#27039)

2023-11-23 15:58:21 +00:00

accelerate.md

Fix typos (#25936)

2023-09-04 11:15:12 +01:00

add_new_model.md

Update add_new_model.md (#26365)

2023-09-25 12:58:11 +02:00

add_new_pipeline.md

Update add_new_pipeline.md (#26197)

2023-09-19 00:41:16 +02:00

add_tensorflow_model.md

Remove utils/documentation_tests.txt (#26213)

2023-09-18 13:33:01 +02:00

attention.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

autoclass_tutorial.md

Update autoclass_tutorial.md (#25929)

2023-09-04 11:16:49 +01:00

benchmarks.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

bertology.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

big_models.md

Fix typos (#25936)

2023-09-04 11:15:12 +01:00

chat_templating.md

Update and reorder docs for chat templates (#27443)

2023-11-14 18:26:13 +00:00

community.md

Update community.md (#25928)

2023-09-04 11:16:34 +01:00

contributing.md

Enable doc in Spanish (#16518)

2022-04-04 10:25:46 -04:00

create_a_model.md

[docs] fixed links with 404 (#27327)

2023-11-06 19:45:03 +00:00

custom_models.md

[docs] Custom model doc update (#27213)

2023-11-03 08:03:13 -04:00

custom_tools.md

[doc] Always call it Agents for consistency (#25958)

2023-09-05 12:27:20 +01:00

debugging.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

fast_tokenizers.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

generation_strategies.md

[docs] navigation improvement between text gen pipelines and text gen params (#26477)

2023-09-29 09:43:39 +02:00

glossary.md

[docs] Performance docs refactor p.2 (#26791)

2023-10-24 13:10:06 -04:00

hpo_train.md

Remove-auth-token (#27060)

2023-11-13 14:20:54 +01:00

index.md

Add UnivNet Vocoder Model for Tortoise TTS Diffusers Integration (#24799)

2023-11-22 17:21:36 +01:00

installation.md

[docs] Update offline mode docs (#26478)

2023-09-29 09:42:21 +02:00

llm_tutorial_optimization.md

Generate: Update docs regarding reusing past_key_values in generate (#27612)

2023-11-21 10:48:14 +00:00

llm_tutorial.md

Generate: update basic llm tutorial (#26937)

2023-10-19 16:53:28 +01:00

model_memory_anatomy.md

Fix typos (#25936)

2023-09-04 11:15:12 +01:00

model_sharing.md

Fix typos (#25936)

2023-09-04 11:15:12 +01:00

model_summary.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

multilingual.md

Fix typo in example code (#25583)

2023-08-18 07:58:59 +02:00

notebooks.md

Enable doc in Spanish (#16518)

2022-04-04 10:25:46 -04:00

pad_truncation.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

peft.md

[Peft] modules_to_save support for peft integration (#27466)

2023-11-14 10:32:57 +01:00

perf_hardware.md

🌐 [i18n-KO] Translated perf_hardware.md to Korean (#24966)

2023-07-25 07:44:24 -04:00

perf_infer_cpu.md

[docs] Update CPU/GPU inference docs (#26881)

2023-10-31 09:44:51 -07:00

perf_infer_gpu_one.md

update Bark FA2 docs (#27400)

2023-11-10 13:40:30 +00:00

perf_torch_compile.md

Fix rendering for torch.compile() docs (#25432)

2023-08-10 13:25:00 +02:00

perf_train_cpu_many.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

perf_train_cpu.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

perf_train_gpu_many.md

[docs] fixed links with 404 (#27327)

2023-11-06 19:45:03 +00:00

perf_train_gpu_one.md

[docs] fixed links with 404 (#27327)

2023-11-06 19:45:03 +00:00

perf_train_special.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

perf_train_tpu_tf.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

perf_train_tpu.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

performance.md

[docs] Update CPU/GPU inference docs (#26881)

2023-10-31 09:44:51 -07:00

perplexity.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

philosophy.md

[docs] fixed links with 404 (#27327)

2023-11-06 19:45:03 +00:00

pipeline_tutorial.md

[ASR Pipe] Improve docs and error messages (#26476)

2023-09-29 18:32:37 +01:00

pipeline_webserver.md

Suggestions on Pipeline_webserver (#25570)

2023-08-18 10:17:44 +02:00

pr_checks.md

Docstring check (#26052)

2023-10-04 15:13:37 +02:00

preprocessing.md

Broken links fixed related to datasets docs (#27569)

2023-11-17 13:44:09 -08:00

quicktour.md

[TYPO] fix typo/format in quicktour.md (#25519)

2023-08-16 08:03:23 +02:00

run_scripts.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

sagemaker.md

[docs] fixed links with 404 (#27327)

2023-11-06 19:45:03 +00:00

serialization.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

task_summary.md

Fix doctest (#25031)

2023-07-25 22:10:06 +02:00

tasks_explained.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

testing.md

Device agnostic testing (#25870)

2023-10-24 16:49:26 +02:00

tf_xla.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

tflite.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

tokenizer_summary.md

Fix typo: Roberta -> RoBERTa (#25302)

2023-08-03 14:17:30 -07:00

torchscript.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

training.md

Broken links fixed related to datasets docs (#27569)

2023-11-17 13:44:09 -08:00

transformers_agents.md

[doc] Always call it Agents for consistency (#25958)

2023-09-05 12:27:20 +01:00

troubleshooting.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00