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

Fix bugs from pipeline preprocessor overhaul (#39425)

Browse Source 
* 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
Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/pipelines/video_classification.py
View File

@@ -51,8 +51,8 @@ class VideoClassificationPipeline(Pipeline):
[huggingface.co/models](https://huggingface.co/models?filter=video-classification). [huggingface.co/models](https://huggingface.co/models?filter=video-classification).
""" """
_load_processor = True _load_processor = False
_load_image_processor = False _load_image_processor = True
_load_feature_extractor = False _load_feature_extractor = False
_load_tokenizer = False _load_tokenizer = False

1
tests/pipelines/test_pipelines_common.py
View File

@@ -846,6 +846,7 @@ class CustomPipelineTest(unittest.TestCase):
"automatic-speech-recognition", "automatic-speech-recognition",
model="hf-internal-testing/fake-custom-wav2vec2", model="hf-internal-testing/fake-custom-wav2vec2",
feature_extractor="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, trust_remote_code=True,
) )