SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Experimental symbolic tracing feature with torch.fx for BERT, ELECTRA and T5 (#11475)

Browse Source 
Symbolic tracing feature for BERT, ELECTRA and T5

Co-authored-by: Michael Benayoun <michael@huggingface.co>
Co-authored-by: Stas Bekman <stas@stason.org>
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
This commit is contained in:
Michael Benayoun
2021-05-14 20:57:30 +02:00
committed by GitHub
parent 94a2348706
commit 86d5fb0b36
7 changed files with 371 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_modeling_t5.py
View File

@@ -488,6 +488,7 @@ class T5ModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCase):
all_model_classes = (T5Model, T5ForConditionalGeneration) if is_torch_available() else ()
all_generative_model_classes = (T5ForConditionalGeneration,) if is_torch_available() else ()
fx_ready_model_classes = all_model_classes
all_parallelizable_model_classes = (T5Model, T5ForConditionalGeneration) if is_torch_available() else ()
test_pruning = False
test_torchscript = True