removes decord (#33987)

* removes decord dependency

optimize

np

Revert "optimize"

This reverts commit faa136b51ec4ec5858e5b0ae40eb7ef89a88b475.

helpers as documentation

pydoc

missing keys

* make fixup

* require_av

---------

Co-authored-by: ad <hi@arnaudiaz.com>
This commit is contained in:
Name
2024-10-17 17:27:34 +02:00
committed by GitHub
parent f2846ad2b7
commit 7f5088503f
9 changed files with 37 additions and 39 deletions

View File

@@ -51,7 +51,7 @@ from transformers.pipelines import (
)
from transformers.testing_utils import (
is_pipeline_test,
require_decord,
require_av,
require_pytesseract,
require_timm,
require_torch,
@@ -722,14 +722,14 @@ class PipelineTesterMixin:
@is_pipeline_test
@require_torch_or_tf
@require_vision
@require_decord
@require_av
def test_pipeline_video_classification(self):
self.run_task_tests(task="video-classification")
@is_pipeline_test
@require_vision
@require_decord
@require_torch
@require_av
def test_pipeline_video_classification_fp16(self):
self.run_task_tests(task="video-classification", torch_dtype="float16")