transformers/docs/source/en/main_classes
Marc Sun e03a9cc0cd
Modify device_map behavior when loading a model using from_pretrained (#23922)
* Modify device map behavior for 4/8 bits model

* Remove device_map arg for training 4/8 bit model

* Remove index

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

* Add Exceptions

* Modify comment

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

* Fix formatting

* Get current device with accelerate

* Revert "Get current device with accelerate"

This reverts commit 46f0079910.

* Fix Exception

* Modify quantization doc

* Fix error

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

---------

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
2023-06-01 13:21:22 -04:00
..
agent.mdx Add local agent (#23438) 2023-05-18 11:09:55 -04:00
callback.mdx Adds a FlyteCallback (#23759) 2023-05-30 10:08:07 -04:00
configuration.mdx
data_collator.mdx
deepspeed.mdx
feature_extractor.mdx
image_processor.mdx
keras_callbacks.mdx
logging.mdx
model.mdx
onnx.mdx
optimizer_schedules.mdx
output.mdx
pipelines.mdx
processors.mdx
quantization.mdx Modify device_map behavior when loading a model using from_pretrained (#23922) 2023-06-01 13:21:22 -04:00
text_generation.mdx
tokenizer.mdx
trainer.mdx Update trainer.mdx class_weights example (#23787) 2023-05-26 08:36:33 -04:00