riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

Fix test_load_default_pipelines_pt for ClapModel (#21886)

Browse Source 
* fix tests

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar 2023-03-01 21:52:26 +01:00 committed by GitHub
parent 36ee128375
commit 89359e4c63
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/pipelines/__init__.py
View File

@ -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": {