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

Update tiny model summary file (#27388)

Browse Source 
* update

* fix

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2023-11-23 21:00:39 +01:00
committed by GitHub
parent fe1c16e95a
commit b8db265bc6
9 changed files with 277 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/models/fuyu/test_modeling_fuyu.py
View File

@@ -24,6 +24,7 @@ from transformers.testing_utils import require_torch, require_torch_gpu, slow, t
from transformers.utils import cached_property
from ...test_modeling_common import ModelTesterMixin, ids_tensor, random_attention_mask
from ...test_pipeline_mixin import PipelineTesterMixin
if is_vision_available():
@@ -262,9 +263,9 @@ class FuyuModelTester:
@require_torch
class FuyuModelTest(ModelTesterMixin, unittest.TestCase):
class FuyuModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
all_model_classes = (FuyuForCausalLM,) if is_torch_available() else ()
pipeline_model_mapping = {"image-to-text": FuyuForCausalLM} if is_torch_available() else {}
pipeline_model_mapping = {"text-generation": FuyuForCausalLM} if is_torch_available() else {}
test_head_masking = False
test_pruning = False