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
..
benchmark	[Test refactor 1/5] Per-folder tests reorganization (#15725)	2022-02-23 15:46:28 -05:00
bettertransformer	Add methods to PreTrainedModel to use PyTorch's BetterTransformer (#21259)	2023-04-27 11:03:42 +02:00
bitsandbytes	Modify device_map behavior when loading a model using from_pretrained (#23922)	2023-06-01 13:21:22 -04:00
deepspeed	accelerate deepspeed and gradient accumulation integrate (#23236)	2023-05-31 15:16:22 +05:30
extended	[tests] switch to torchrun (#22712)	2023-04-12 08:25:45 -07:00
fixtures	[WIP] add SpeechT5 model (#18922)	2023-02-03 12:43:46 -05:00
generation	Better TF docstring types (#23477)	2023-05-24 13:52:52 +01:00
models	fix: Replace add_prefix_space in get_prompt_ids with manual space for FastTokenizer compatibility (#23796)	2023-05-31 10:52:35 -04:00
onnx	Fix issue introduced in PR #23163 (#23363)	2023-05-15 11:38:44 +02:00
optimization	Make schedulers picklable by making lr_lambda fns global (#21768)	2023-03-02 12:08:43 -05:00
pipelines	[image-to-text pipeline] Add conditional text support + GIT (#23362)	2023-05-22 21:45:50 +02:00
repo_utils	Revamp test selection for the example tests (#23737)	2023-05-25 09:38:21 -04:00
sagemaker	Avoid invalid escape sequences, use raw strings (#22936)	2023-04-25 09:17:56 -04:00
tokenization	Update quality tooling for formatting (#21480)	2023-02-06 18:10:56 -05:00
tools	Fix image segmentation tool test (#23306)	2023-05-11 14:38:11 -04:00
trainer	Paged Optimizer + Lion Optimizer for Trainer (#23217)	2023-05-24 12:53:28 +02:00
utils	Fix the regex in get_imports to support multiline try blocks and excepts with specific exception types (#23725)	2023-05-24 15:40:19 -04:00
__init__.py	GPU text generation: mMoved the encoded_prompt to correct device	2020-01-06 15:11:12 +01:00
test_backbone_common.py	Add methods to update and verify out_features out_indices (#23031)	2023-05-04 10:15:06 +01:00
test_configuration_common.py	Add RWKV-4 (#22797)	2023-05-09 13:04:10 -04:00
test_feature_extraction_common.py	Prepare tests for hfh 0.14 (#22958)	2023-04-24 09:31:50 -04:00
test_image_processing_common.py	Prepare tests for hfh 0.14 (#22958)	2023-04-24 09:31:50 -04:00
test_image_transforms.py	Bug fix - flip_channel_order for channels first images (#23701)	2023-05-31 17:12:27 +01:00
test_modeling_common.py	[AutoModel] fix torch_dtype=auto in from_pretrained (#23379)	2023-05-16 10:21:42 -07:00
test_modeling_flax_common.py	Fix bigbird random attention (#21023)	2023-04-27 13:52:28 -04:00
test_modeling_tf_common.py	Overhaul TF serving signatures + dummy inputs (#23234)	2023-05-24 17:03:24 +01:00
test_pipeline_mixin.py	Update tiny models and pipeline tests (#23446)	2023-05-18 17:29:04 +02:00
test_sequence_feature_extraction_common.py	Apply ruff flake8-comprehensions (#21694)	2023-02-22 09:14:54 +01:00
test_tokenization_common.py	chore: allow protobuf 3.20.3 requirement (#22759)	2023-05-10 20:22:56 +02:00