Fix bugs from pipeline preprocessor overhaul (#39425)

* Correct load classes for VideoClassificationPipeline

* Correct load classes for the ASR pipeline
This commit is contained in:
Matt
2025-07-15 14:28:59 +01:00
committed by GitHub
parent 53c9dcd6fd
commit a989bf8d84
2 changed files with 3 additions and 2 deletions

View File

@@ -846,6 +846,7 @@ class CustomPipelineTest(unittest.TestCase):
"automatic-speech-recognition",
model="hf-internal-testing/fake-custom-wav2vec2",
feature_extractor="hf-internal-testing/fake-custom-wav2vec2",
tokenizer="facebook/wav2vec2-base-960h", # Test workaround - the pipeline requires a tokenizer
trust_remote_code=True,
)