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:
@@ -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")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user