transformers/docs/source/en/main_classes
Sylvain Gugger 56f50590d5
Use Accelerate in from_pretrained for big model inference (#17341)
* Initial work

* More or less finished with first draft

* Update src/transformers/modeling_utils.py

Co-authored-by: Stas Bekman <stas00@users.noreply.github.com>

* Update src/transformers/modeling_utils.py

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

* Fix randomly initialized weights

* Update src/transformers/modeling_utils.py

Co-authored-by: Lysandre Debut <lysandre.debut@reseau.eseo.fr>

* Address review comments

* Rename DeepSpeed folder to temporarily fix the test issue?

* Revert to try if Accelerate fix works

* Use latest Accelerate release

* Quality and fixes

* Style

* Quality

* Add doc

* Test + fix

* More blocks

Co-authored-by: Stas Bekman <stas00@users.noreply.github.com>
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
Co-authored-by: Lysandre Debut <lysandre.debut@reseau.eseo.fr>
2022-05-23 14:32:21 -04:00
..
callback.mdx Enable doc in Spanish (#16518) 2022-04-04 10:25:46 -04:00
configuration.mdx Enable doc in Spanish (#16518) 2022-04-04 10:25:46 -04:00
data_collator.mdx Enable doc in Spanish (#16518) 2022-04-04 10:25:46 -04:00
deepspeed.mdx Enable doc in Spanish (#16518) 2022-04-04 10:25:46 -04:00
feature_extractor.mdx Enable doc in Spanish (#16518) 2022-04-04 10:25:46 -04:00
keras_callbacks.mdx Enable doc in Spanish (#16518) 2022-04-04 10:25:46 -04:00
logging.mdx logging documentation update (#17174) 2022-05-16 16:47:28 -04:00
model.mdx Use Accelerate in from_pretrained for big model inference (#17341) 2022-05-23 14:32:21 -04:00
onnx.mdx Enable doc in Spanish (#16518) 2022-04-04 10:25:46 -04:00
optimizer_schedules.mdx Enable doc in Spanish (#16518) 2022-04-04 10:25:46 -04:00
output.mdx Add Wav2Vec2Conformer (#16812) 2022-05-17 00:43:16 +02:00
pipelines.mdx Enable doc in Spanish (#16518) 2022-04-04 10:25:46 -04:00
processors.mdx Enable doc in Spanish (#16518) 2022-04-04 10:25:46 -04:00
text_generation.mdx Enable doc in Spanish (#16518) 2022-04-04 10:25:46 -04:00
tokenizer.mdx documentation: some minor clean up (#16850) 2022-04-26 16:56:08 -04:00
trainer.mdx PyTorch FSDP integration in Trainer (#17136) 2022-05-09 20:40:56 +05:30