Fix pipeline test

huggingface · Nov 21, 2022 · 3d29faa · 3d29faa
1 parent d2fc0ae
commit 3d29faa
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/pipelines/test_pipelines_common.py b/tests/pipelines/test_pipelines_common.py
@@ -157,7 +157,9 @@ def get_tiny_feature_extractor_from_checkpoint(checkpoint, tiny_config, feature_
 
     # Audio Spectogram Transformer specific.
     if feature_extractor.__class__.__name__ == "ASTFeatureExtractor":
-        feature_extractor = feature_extractor.__class__(max_length=24, num_mel_bins=16)
+        feature_extractor = feature_extractor.__class__(
+            max_length=tiny_config.max_length, num_mel_bins=tiny_config.num_mel_bins
+        )
 
     # Speech2TextModel specific.
     if hasattr(tiny_config, "input_feat_per_channel") and feature_extractor: