fix checkpoint (#21874)

43299c63 · Arthur · GitHub · 89359e4c · 43299c63
Unverified Commit 43299c63 authored Mar 02, 2023 by Arthur Committed by GitHub Mar 02, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/pipelines/zero_shot_audio_classification.py src/transformers/pipelines/zero_shot_audio_classification.py +1 -1

No files found.
--- a/src/transformers/pipelines/zero_shot_audio_classification.py
+++ b/src/transformers/pipelines/zero_shot_audio_classification.py
@@ -32,7 +32,7 @@ class ZeroShotAudioClassificationPipeline(ChunkPipeline):

    >>> dataset = load_dataset("ashraq/esc50")
    >>> audio = next(iter(dataset["train"]["audio"]))["array"]
-    >>> classifier = pipeline(task="zero-shot-audio-classification", model="laion-ai/clap-htsat-tiny")
+    >>> classifier = pipeline(task="zero-shot-audio-classification", model="laion/clap-htsat-unfused")
    >>> classifier(audio, candidate_labels=["Sound of a dog", "Sound of vaccum cleaner"])
    ```