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

[TTA Pipeline] Test MusicGen and VITS (#26146)

Browse Source
This commit is contained in:
Sanchit Gandhi
2023-09-15 10:00:36 +01:00
committed by GitHub
parent 869733ab62
commit d70fab8b20
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/musicgen/test_modeling_musicgen.py
View File

@@ -502,7 +502,7 @@ class MusicgenTester:
class MusicgenTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixin, unittest.TestCase): class MusicgenTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixin, unittest.TestCase):
all_model_classes = (MusicgenForConditionalGeneration,) if is_torch_available() else () all_model_classes = (MusicgenForConditionalGeneration,) if is_torch_available() else ()
greedy_sample_model_classes = (MusicgenForConditionalGeneration,) if is_torch_available() else () greedy_sample_model_classes = (MusicgenForConditionalGeneration,) if is_torch_available() else ()
pipeline_model_mapping = {} pipeline_model_mapping = {"text-to-audio": MusicgenForConditionalGeneration} if is_torch_available() else {}
test_pruning = False # training is not supported yet for MusicGen test_pruning = False # training is not supported yet for MusicGen
test_headmasking = False test_headmasking = False
test_resize_embeddings = False test_resize_embeddings = False

4
tests/models/vits/test_modeling_vits.py
View File

@@ -40,6 +40,7 @@ from ...test_modeling_common import (
ids_tensor, ids_tensor,
random_attention_mask, random_attention_mask,
) )
from ...test_pipeline_mixin import PipelineTesterMixin
if is_torch_available(): if is_torch_available():
@@ -153,8 +154,9 @@ class VitsModelTester:
@require_torch @require_torch
class VitsModelTest(ModelTesterMixin, unittest.TestCase): class VitsModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
all_model_classes = (VitsModel,) if is_torch_available() else () all_model_classes = (VitsModel,) if is_torch_available() else ()
pipeline_model_mapping = {"text-to-audio": VitsModel} if is_torch_available() else {}
is_encoder_decoder = False is_encoder_decoder = False
test_pruning = False test_pruning = False
test_headmasking = False test_headmasking = False