Fix bugs from pipeline preprocessor overhaul (#39425)
* Correct load classes for VideoClassificationPipeline * Correct load classes for the ASR pipeline
This commit is contained in:
@@ -51,8 +51,8 @@ class VideoClassificationPipeline(Pipeline):
|
||||
[huggingface.co/models](https://huggingface.co/models?filter=video-classification).
|
||||
"""
|
||||
|
||||
_load_processor = True
|
||||
_load_image_processor = False
|
||||
_load_processor = False
|
||||
_load_image_processor = True
|
||||
_load_feature_extractor = False
|
||||
_load_tokenizer = False
|
||||
|
||||
|
||||
@@ -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,
|
||||
)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user