mirror of
https://github.com/saymrwulf/transformers.git
synced 2026-05-15 21:01:19 +00:00
b67fd797be
* Start the work on TFVisionEncoderDecoderModel * Expose TFVisionEncoderDecoderModel * fix import * Add modeling_tf_vision_encoder_decoder to _ignore_modules in get_model_modules() * reorder * Apply the fix for checkpoint loading as in #14016 * remove attention_mask + fix VISION_DUMMY_INPUTS * A minimal change to make TF generate() work for vision models as encoder in encoder-decoder setting * fix wrong condition: shape_list(input_ids) == 2 * add tests * use personal TFViTModel checkpoint (for now) * Add equivalence tests + projection layer * style * make sure projection layer can run * Add examples * Apply suggestions from code review Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> * Clean comments (need to work on TODOs for PyTorch models) * Remove TF -> PT in check_pt_tf_equivalence for TFVisionEncoderDecoderModel * fixes * Revert changes in PT code. * Update tests/test_modeling_tf_vision_encoder_decoder.py Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * Add test_inference_coco_en for TF test * fix quality * fix name * build doc * add main_input_name * Fix ckpt name in test * fix diff between master and this PR * fix doc * fix style and quality * fix missing doc * fix labels handling * Delete auto.rst * Add the changes done in #14016 * fix prefix * Apply suggestions from code review Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> * make style Co-authored-by: ydshieh <ydshieh@users.noreply.github.com> Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> |
||
|---|---|---|
| .. | ||
| benchmark | ||
| commands | ||
| data | ||
| models | ||
| onnx | ||
| pipelines | ||
| sagemaker | ||
| utils | ||
| __init__.py | ||
| activations.py | ||
| activations_tf.py | ||
| configuration_utils.py | ||
| convert_graph_to_onnx.py | ||
| convert_pytorch_checkpoint_to_tf2.py | ||
| convert_slow_tokenizer.py | ||
| convert_slow_tokenizers_checkpoints_to_fast.py | ||
| convert_tf_hub_seq_to_seq_bert_to_pytorch.py | ||
| debug_utils.py | ||
| deepspeed.py | ||
| dependency_versions_check.py | ||
| dependency_versions_table.py | ||
| feature_extraction_sequence_utils.py | ||
| feature_extraction_utils.py | ||
| file_utils.py | ||
| generation_beam_search.py | ||
| generation_flax_logits_process.py | ||
| generation_flax_utils.py | ||
| generation_logits_process.py | ||
| generation_stopping_criteria.py | ||
| generation_tf_utils.py | ||
| generation_utils.py | ||
| hf_argparser.py | ||
| image_utils.py | ||
| integrations.py | ||
| keras_callbacks.py | ||
| modelcard.py | ||
| modeling_flax_outputs.py | ||
| modeling_flax_pytorch_utils.py | ||
| modeling_flax_utils.py | ||
| modeling_outputs.py | ||
| modeling_tf_outputs.py | ||
| modeling_tf_pytorch_utils.py | ||
| modeling_tf_utils.py | ||
| modeling_utils.py | ||
| optimization.py | ||
| optimization_tf.py | ||
| py.typed | ||
| testing_utils.py | ||
| tokenization_utils.py | ||
| tokenization_utils_base.py | ||
| tokenization_utils_fast.py | ||
| trainer.py | ||
| trainer_callback.py | ||
| trainer_pt_utils.py | ||
| trainer_seq2seq.py | ||
| trainer_tf.py | ||
| trainer_utils.py | ||
| training_args.py | ||
| training_args_seq2seq.py | ||
| training_args_tf.py | ||