SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix Fatal Python error: Bus error in ZeroShotAudioClassificationPipelineTests (#30283)

Browse Source 
fix

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2024-04-17 11:47:30 +02:00
committed by GitHub
parent 05dab4e5ba
commit eb75516e7c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/pipelines/test_pipelines_zero_shot_audio_classification.py
View File

@@ -32,7 +32,7 @@ class ZeroShotAudioClassificationPipelineTests(unittest.TestCase):
audio_classifier = pipeline( audio_classifier = pipeline(
task="zero-shot-audio-classification", model="hf-internal-testing/tiny-clap-htsat-unfused" task="zero-shot-audio-classification", model="hf-internal-testing/tiny-clap-htsat-unfused"
) )
dataset = load_dataset("ashraq/esc50") dataset = load_dataset("hf-internal-testing/ashraq-esc50-1-dog-example")
audio = dataset["train"]["audio"][-1]["array"] audio = dataset["train"]["audio"][-1]["array"]
output = audio_classifier(audio, candidate_labels=["Sound of a dog", "Sound of vaccum cleaner"]) output = audio_classifier(audio, candidate_labels=["Sound of a dog", "Sound of vaccum cleaner"])
self.assertEqual( self.assertEqual(
@@ -52,7 +52,7 @@ class ZeroShotAudioClassificationPipelineTests(unittest.TestCase):
model="laion/clap-htsat-unfused", model="laion/clap-htsat-unfused",
) )
# This is an audio of a dog # This is an audio of a dog
dataset = load_dataset("ashraq/esc50") dataset = load_dataset("hf-internal-testing/ashraq-esc50-1-dog-example")
audio = dataset["train"]["audio"][-1]["array"] audio = dataset["train"]["audio"][-1]["array"]
output = audio_classifier(audio, candidate_labels=["Sound of a dog", "Sound of vaccum cleaner"]) output = audio_classifier(audio, candidate_labels=["Sound of a dog", "Sound of vaccum cleaner"])