Nicolas Patry ccc0897804 Adding support for device_map directly in pipeline(..) function. (#17902) ... * Adding support for `device_map` directly in `pipeline(..)` function. * Updating the docstring. * Adding a better docstring * Put back type hints. * Blacked. (`make fixup` didn't work ??!!)		2022-07-15 15:54:26 +02:00
..
benchmark	[Test refactor 1/5] Per-folder tests reorganization (#15725)	2022-02-23 15:46:28 -05:00
deepspeed	deprecate is_torch_bf16_available (#17738)	2022-06-20 08:40:11 -04:00
extended	Update self-push workflow (#17177)	2022-05-13 16:28:00 +02:00
fixtures	add a warning in SpmConverter for sentencepiece's model using the byte fallback feature (#16629)	2022-04-11 11:06:10 +02:00
generation	Update expected values in constrained beam search tests (#17887)	2022-06-28 08:53:53 +02:00
models	Add TF DeiT implementation (#17806)	2022-07-13 18:04:08 +01:00
onnx	Squash commits (#17981)	2022-07-06 08:11:48 -04:00
optimization	[Test refactor 1/5] Per-folder tests reorganization (#15725)	2022-02-23 15:46:28 -05:00
pipelines	Adding support for device_map directly in pipeline(..) function. (#17902)	2022-07-15 15:54:26 +02:00
sagemaker	Black preview (#17217)	2022-05-12 16:25:55 -04:00
tokenization	fix train_new_from_iterator in the case of byte-level tokenizers (#17549)	2022-06-08 15:30:41 +02:00
trainer	Enable torchdynamo with torch_tensorrt(fx path) (#17765)	2022-07-13 12:43:28 -04:00
utils	Make sharded checkpoints work in offline mode (#18125)	2022-07-13 12:43:08 -04:00
__init__.py	GPU text generation: mMoved the encoded_prompt to correct device	2020-01-06 15:11:12 +01:00
test_configuration_common.py	XLA train step fixes (#17973)	2022-07-01 19:11:14 +01:00
test_feature_extraction_common.py	Compute min_resolution in prepare_image_inputs (#17915)	2022-06-29 10:30:20 +02:00
test_modeling_common.py	Offload fixes (#17810)	2022-06-22 12:23:07 -04:00
test_modeling_flax_common.py	[Flax] Add remat (gradient checkpointing) (#17843)	2022-07-01 18:33:54 +01:00
test_modeling_tf_common.py	TF: unpack_inputs decorator independent from main_input_name (#18110)	2022-07-13 10:43:41 +01:00
test_sequence_feature_extraction_common.py	Some tests misusing assertTrue for comparisons fix (#16771)	2022-04-19 14:44:08 +02:00
test_tokenization_common.py	Fix some typos. (#17560)	2022-07-11 05:00:13 -04:00