mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-27 00:09:00 +06:00
de4159a318
* Add a test to ensure int dummy inputs are int64 * Move the test into the existing int64 test and update a lot of existing dummies * Fix remaining dummies * Fix remaining dummies * Test for int64 serving sigs as well * Update core tests to use tf.int64 * Add better messages to the assertions * Update all serving sigs to int64 * More sneaky hiding tf.int32s * Add an optional int32 signature in save_pretrained * make fixup * Add Amy's suggestions * Switch all serving sigs back to tf.int32 * Switch all dummies to tf.int32 * Adjust tests to check for tf.int32 instead of tf.int64 * Fix base dummy_inputs dtype * Start casting to tf.int32 in input_processing * Change dtype for unpack_inputs test * Add proper tf.int32 test * Make the alternate serving signature int64 |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| {{cookiecutter.lowercase_modelname}}.mdx | ||
| configuration_{{cookiecutter.lowercase_modelname}}.py | ||
| configuration.json | ||
| modeling_{{cookiecutter.lowercase_modelname}}.py | ||
| modeling_flax_{{cookiecutter.lowercase_modelname}}.py | ||
| modeling_tf_{{cookiecutter.lowercase_modelname}}.py | ||
| test_modeling_{{cookiecutter.lowercase_modelname}}.py | ||
| test_modeling_flax_{{cookiecutter.lowercase_modelname}}.py | ||
| test_modeling_tf_{{cookiecutter.lowercase_modelname}}.py | ||
| to_replace_{{cookiecutter.lowercase_modelname}}.py | ||
| tokenization_{{cookiecutter.lowercase_modelname}}.py | ||
| tokenization_fast_{{cookiecutter.lowercase_modelname}}.py | ||