mirror of
https://github.com/saymrwulf/transformers.git
synced 2026-05-15 21:01:19 +00:00
8e8384663d
* Add serving_output and serving methods to some vision models * Add serving outputs for DeiT * Don't convert hidden states - differing shapes * Make saveable * Fix up * Make swin saveable * Add in tests * Fix funnel tests (can't convert to tensor) * Fix numpy call * Tidy up a bit * Add in hidden states - resnet * Remove numpy * Fix failing tests - tensor shape and skipping tests * Remove duplicated function * PR comments - formatting and var names * PR comments Add suggestions made by Joao Gante: * Use tf.shape instead of shape_list * Use @tooslow decorator on tests * Simplify some of the logic * PR comments Address Yih-Dar Sheih comments - making tensor names consistent and make types float * Types consistent with docs; disable test on swin (slow) * CI trigger * Change input_features to float32 * Add serving_output for segformer * Fixup Co-authored-by: Amy Roberts <amyeroberts@users.noreply.github.com> |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| test_activations.py | ||
| test_activations_tf.py | ||
| test_add_new_model_like.py | ||
| test_cli.py | ||
| test_convert_slow_tokenizer.py | ||
| test_doc_samples.py | ||
| test_file_utils.py | ||
| test_generic.py | ||
| test_hf_argparser.py | ||
| test_image_utils.py | ||
| test_logging.py | ||
| test_model_card.py | ||
| test_model_output.py | ||
| test_modeling_tf_core.py | ||
| test_offline.py | ||
| test_skip_decorators.py | ||
| test_utils_check_copies.py | ||
| test_versions_utils.py | ||