transformers/src/transformers
Zach Mueller 6ad9c8f743
Non blocking support to torch DL's (#30465)
* Non blocking support

* Check for optimization

* Doc
2024-04-24 16:24:23 -04:00
..
benchmark Make torch xla available on GPU (#29334) 2024-03-11 14:07:16 +00:00
commands Remove add-new-model in favor of add-new-model-like (#30424) 2024-04-24 09:38:18 +02:00
data Remove fast tokenization warning in Data Collators (#28213) 2024-01-02 18:32:23 +00:00
generation Make EosTokenCriteria compatible with mps (#30376) 2024-04-23 15:23:52 +02:00
integrations [FEAT]: EETQ quantizer support (#30262) 2024-04-22 20:38:58 +01:00
kernels DeformableDETR support bfloat16 (#29232) 2024-03-04 14:18:09 +00:00
models Phi-3 (#30423) 2024-04-24 17:32:09 +02:00
onnx Byebye torch 1.10 (#28207) 2024-01-11 16:18:27 +01:00
pipelines Update docstrings for text generation pipeline (#30343) 2024-04-22 14:01:30 +01:00
quantizers [FEAT]: EETQ quantizer support (#30262) 2024-04-22 20:38:58 +01:00
sagemaker
tools Warn about tool use (#29628) 2024-03-13 14:53:13 +01:00
utils Phi-3 (#30423) 2024-04-24 17:32:09 +02:00
__init__.py Phi-3 (#30423) 2024-04-24 17:32:09 +02:00
activations_tf.py Add tf_keras imports to prepare for Keras 3 (#28588) 2024-01-30 17:26:36 +00:00
activations.py v4.39 deprecations 🧼 (#29492) 2024-03-07 10:44:43 +00:00
audio_utils.py Add MusicGen Melody (#28819) 2024-03-18 13:06:12 +00:00
cache_utils.py Generate: handle cache_position update in generate (#29467) 2024-03-14 16:35:31 +00:00
configuration_utils.py Fix attn_implementation documentation (#29295) 2024-02-27 10:43:01 +01:00
convert_graph_to_onnx.py Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
convert_pytorch_checkpoint_to_tf2.py Remove static pretrained maps from the library's internals (#29112) 2024-03-25 10:33:38 +01:00
convert_slow_tokenizer.py [Llava] + CIs fix red cis and llava integration tests (#30440) 2024-04-24 10:51:35 +02:00
convert_slow_tokenizers_checkpoints_to_fast.py
convert_tf_hub_seq_to_seq_bert_to_pytorch.py Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
debug_utils.py
deepspeed.py
dependency_versions_check.py
dependency_versions_table.py Jax: scipy version pin (#30402) 2024-04-23 10:42:17 +01:00
dynamic_module_utils.py Reset alarm signal when the function is ended (#29706) 2024-03-30 17:41:27 +01:00
feature_extraction_sequence_utils.py
feature_extraction_utils.py [Feature Extractors] Fix kwargs to pre-trained (#30260) 2024-04-19 11:16:08 +01:00
file_utils.py Make torch xla available on GPU (#29334) 2024-03-11 14:07:16 +00:00
hf_argparser.py Add test for parse_json_file and change typing to os.PathLike (#30183) 2024-04-15 09:34:36 +01:00
hyperparameter_search.py [integration] Update Ray Tune integration for Ray 2.7 (#26499) 2023-12-09 11:04:13 +01:00
image_processing_utils.py [LlavaNext] Fix llava next unsafe imports (#29773) 2024-03-21 13:47:58 +01:00
image_transforms.py Add Idefics2 (#30253) 2024-04-15 17:03:03 +01:00
image_utils.py Add Idefics2 (#30253) 2024-04-15 17:03:03 +01:00
keras_callbacks.py Add tf_keras imports to prepare for Keras 3 (#28588) 2024-01-30 17:26:36 +00:00
modelcard.py Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
modeling_attn_mask_utils.py FIX: Fixes unexpected behaviour for Llava / LLama & AWQ Fused modules + revert #30070 at the same time (#30317) 2024-04-18 15:51:17 +02:00
modeling_flax_outputs.py
modeling_flax_pytorch_utils.py Flax mistral (#26943) 2024-01-31 14:19:02 +01:00
modeling_flax_utils.py [Gemma] final fixes to the modeling (#29729) 2024-03-19 14:47:42 +01:00
modeling_outputs.py Improved type hinting for all attention parameters (#28479) 2024-01-24 16:47:34 +00:00
modeling_tf_outputs.py
modeling_tf_pytorch_utils.py Cast bfloat16 to float32 for Numpy conversions (#29755) 2024-03-21 14:04:11 +00:00
modeling_tf_utils.py Support sharded safetensors in TF (#29350) 2024-03-20 14:22:35 +00:00
modeling_utils.py fix for itemsize => element_size() for torch backwards compat (#30133) 2024-04-23 15:00:28 +02:00
optimization_tf.py Fix Keras scheduler import so it works for older versions of Keras (#28895) 2024-02-07 12:28:24 +00:00
optimization.py Fix layerwise GaLore optimizer hard to converge with warmup scheduler (#30372) 2024-04-22 17:00:26 +01:00
processing_utils.py Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
pytorch_utils.py Enable fp16 on CPU (#30459) 2024-04-24 15:38:52 -04:00
safetensors_conversion.py Send headers when converting safetensors (#30144) 2024-04-09 17:03:36 +02:00
testing_utils.py [FEAT]: EETQ quantizer support (#30262) 2024-04-22 20:38:58 +01:00
tf_utils.py Wrap Keras methods to support BatchEncoding (#28734) 2024-01-31 13:18:42 +00:00
time_series_utils.py
tokenization_utils_base.py Deprecate default chat templates (#30346) 2024-04-19 15:41:26 +01:00
tokenization_utils_fast.py
tokenization_utils.py Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
trainer_callback.py 🚨🚨🚨Deprecate evaluation_strategy to eval_strategy🚨🚨🚨 (#30190) 2024-04-18 12:49:43 -04:00
trainer_pt_utils.py Non blocking support to torch DL's (#30465) 2024-04-24 16:24:23 -04:00
trainer_seq2seq.py Trainer: fail early in the presence of an unsavable generation_config (#29675) 2024-03-15 12:59:10 +00:00
trainer_utils.py add Cambricon MLUs support (#29627) 2024-03-27 15:54:28 +01:00
trainer.py Non blocking support to torch DL's (#30465) 2024-04-24 16:24:23 -04:00
training_args_seq2seq.py Update all references to canonical models (#29001) 2024-02-16 08:16:58 +01:00
training_args_tf.py 🚨🚨🚨Deprecate evaluation_strategy to eval_strategy🚨🚨🚨 (#30190) 2024-04-18 12:49:43 -04:00
training_args.py Enable fp16 on CPU (#30459) 2024-04-24 15:38:52 -04:00