[Audio] Qwen Audio Example #1082
Draft
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Signed-off-by: Kyle Sayers <[email protected]>
Signed-off-by: Kyle Sayers <[email protected]>
Signed-off-by: Kyle Sayers <[email protected]>
Signed-off-by: Kyle Sayers <[email protected]>
Signed-off-by: Kyle Sayers <[email protected]>
Signed-off-by: Kyle Sayers <[email protected]>
Signed-off-by: Kyle Sayers <[email protected]>
Signed-off-by: Kyle Sayers <[email protected]>
Signed-off-by: Kyle Sayers <[email protected]>
Signed-off-by: Kyle Sayers <[email protected]>
Signed-off-by: Kyle Sayers <[email protected]>
Signed-off-by: Kyle Sayers <[email protected]>
Signed-off-by: Kyle Sayers <[email protected]>
Signed-off-by: Kyle Sayers <[email protected]>
Signed-off-by: Kyle Sayers <[email protected]>
Signed-off-by: Kyle Sayers <[email protected]>
Signed-off-by: Kyle Sayers <[email protected]>
mgoin
pushed a commit
that referenced
this pull request
Jan 22, 2025
## Purpose ## * Support oneshot with audio datasets ## Changes ## * Extend `apply_pad_mask_to_batch` to handle cases where there are no `input_ids` and where there might be `decoder_input_ids` * Extend `TextGenerationDataset` to detect if a dataset is already tokenized based on `processor.model_input_names` rather than only `input_ids` ## Testing ## * Ran `test_processors.py` to completion, which verifies that the `model_input_names` attribute is defined for most processors * Ran whisper to completion in #1082 <details><summary>test_processors.py</summary> ```python3 import pytest from transformers import AutoProcessor @pytest.mark.parametrize( "model_id,expected", [ ("meta-llama/Meta-Llama-3-8B-Instruct", ["input_ids", "attention_mask"]), ("mistralai/Mixtral-8x7B-Instruct-v0.1", ["input_ids", "attention_mask"]), ( "Qwen/Qwen2-VL-2B-Instruct", [ "input_ids", "attention_mask", "pixel_values", "image_grid_thw", "pixel_values_videos", "video_grid_thw", ], ), ("mgoin/pixtral-12b", ["input_ids", "attention_mask", "pixel_values"]), ("openai/whisper-large-v2", ["input_features"]), ( "Qwen/Qwen2-Audio-7B-Instruct", ["input_ids", "attention_mask", "input_features", "feature_attention_mask"], ), ], ) def test_processor_model_input_names(model_id, expected): """ Tests the model_input_names attribute of common model processors """ processor = AutoProcessor.from_pretrained(model_id, trust_remote_code=True) assert processor.model_input_names == expected ``` </details> --------- Signed-off-by: Kyle Sayers <[email protected]> Co-authored-by: Dipika Sikka <[email protected]>
kylesayrs
changed the title
Signed-off-by: Kyle Sayers <[email protected]>
kylesayrs
force-pushed
the
kylesayrs/audio_examples
branch
from
8897342 to
f1bd1d2
Compare
Signed-off-by: Kyle Sayers <[email protected]>
Signed-off-by: Kyle Sayers <[email protected]>
Signed-off-by: Kyle Sayers <[email protected]>
kylesayrs
changed the base branch from
main
to
kylesayrs/whisper_audio_example
kylesayrs
changed the base branch from
kylesayrs/whisper_audio_example
to
main
kylesayrs
changed the title
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Purpose
Prerequisites
Changes
examples/multimodal_audio/whisper_example.pyexamples/multimodal_audio/qwen2_audio_example.pyTraceableWhisperForConditionalGenerationTraceableQwen2AudioForConditionalGeneration**kwargs, as is the case for the Whisper processor -_-TODO
Testing