Tapas tf (#13393)
* TF Tapas first commit * updated docs * updated logger message * updated pytorch weight conversion script to support scalar array * added use_cache to tapas model config to work properly with tf input_processing * 1. rm embeddings_sum 2. added # Copied 3. + TFTapasMLMHead 4. and lot other small fixes * updated docs * + test for tapas * updated testing_utils to check is_tensorflow_probability_available * converted model logits post processing using numpy to work with both PT and TF models * + TFAutoModelForTableQuestionAnswering * added TF support * added test for TFAutoModelForTableQuestionAnswering * added test for TFAutoModelForTableQuestionAnswering pipeline * updated auto model docs * fixed typo in import * added tensorflow_probability to run tests * updated MLM head * updated tapas.rst with TF model docs * fixed optimizer import in docs * updated convert to np data from pt model is not `transformers.tokenization_utils_base.BatchEncoding` after pipeline upgrade * updated pipeline: 1. with torch.no_gard removed, pipeline forward handles 2. token_type_ids converted to numpy * updated docs. * removed `use_cache` from config * removed floats_tensor * updated code comment * updated Copyright Year and logits_aggregation Optional * updated docs and comments * updated docstring * fixed model weight loading * make fixup * fix indentation * added tf slow pipeline test * pip upgrade * upgrade python to 3.7 * removed from_pt from tests * revert commit f18cfa9
This commit is contained in:
@@ -44,6 +44,7 @@ from .file_utils import (
|
||||
is_scatter_available,
|
||||
is_sentencepiece_available,
|
||||
is_soundfile_availble,
|
||||
is_tensorflow_probability_available,
|
||||
is_tf_available,
|
||||
is_timm_available,
|
||||
is_tokenizers_available,
|
||||
@@ -292,6 +293,19 @@ def require_torch_scatter(test_case):
|
||||
return test_case
|
||||
|
||||
|
||||
def require_tensorflow_probability(test_case):
|
||||
"""
|
||||
Decorator marking a test that requires TensorFlow probability.
|
||||
|
||||
These tests are skipped when TensorFlow probability isn't installed.
|
||||
|
||||
"""
|
||||
if not is_tensorflow_probability_available():
|
||||
return unittest.skip("test requires TensorFlow probability")(test_case)
|
||||
else:
|
||||
return test_case
|
||||
|
||||
|
||||
def require_torchaudio(test_case):
|
||||
"""
|
||||
Decorator marking a test that requires torchaudio. These tests are skipped when torchaudio isn't installed.
|
||||
|
||||
Reference in New Issue
Block a user