From 89359e4c63ac47577edb45c4cc6e5edf908d5190 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Wed, 1 Mar 2023 21:52:26 +0100 Subject: [PATCH] Fix `test_load_default_pipelines_pt` for `ClapModel` (#21886) * fix tests --------- Co-authored-by: ydshieh --- src/transformers/pipelines/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/pipelines/__init__.py b/src/transformers/pipelines/__init__.py index 055d9d4b69..ccd039f7f0 100755 --- a/src/transformers/pipelines/__init__.py +++ b/src/transformers/pipelines/__init__.py @@ -302,7 +302,7 @@ SUPPORTED_TASKS = { }, "zero-shot-audio-classification": { "impl": ZeroShotAudioClassificationPipeline, - "tf": (TFAutoModel,) if is_tf_available() else (), + "tf": (), "pt": (AutoModel,) if is_torch_available() else (), "default": { "model": {