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

Update tiny models for pipeline testing. (#24364)

Browse Source 
* fix

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2023-06-20 14:43:10 +02:00
committed by GitHub
parent 56efbf4301
commit c23d131eab
15 changed files with 110 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/models/autoformer/test_modeling_autoformer.py
View File

@@ -25,6 +25,7 @@ from transformers.testing_utils import require_torch, slow, torch_device
from ...test_configuration_common import ConfigTester
from ...test_modeling_common import ModelTesterMixin, floats_tensor, ids_tensor
from ...test_pipeline_mixin import PipelineTesterMixin
TOLERANCE = 1e-4
@@ -201,9 +202,10 @@ class AutoformerModelTester:
@require_torch
class AutoformerModelTest(ModelTesterMixin, unittest.TestCase):
class AutoformerModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
all_model_classes = (AutoformerModel, AutoformerForPrediction) if is_torch_available() else ()
all_generative_model_classes = (AutoformerForPrediction,) if is_torch_available() else ()
pipeline_model_mapping = {"feature-extraction": AutoformerModel} if is_torch_available() else {}
test_pruning = False
test_head_masking = False
test_missing_keys = False