Skip to content

Fix SpeechT5 generation pipeline for compatibility with transformers==4.55#2308

Merged
regisss merged 7 commits into
huggingface:mainfrom
HabanaAI:main_modeling_speecht5_transformer_4_55_fix
Oct 21, 2025
Merged

Fix SpeechT5 generation pipeline for compatibility with transformers==4.55#2308
regisss merged 7 commits into
huggingface:mainfrom
HabanaAI:main_modeling_speecht5_transformer_4_55_fix

Conversation

@gplutop7
Copy link
Copy Markdown
Collaborator

This PR updates the Habana implementation of SpeechT5 to ensure full compatibility with the transformers==4.55.4 API, aligning the model’s text-to-speech generation path with the upstream behavior while maintaining HPU-specific optimizations.

Validation performed using the Habana Gaudi pipeline:
PT_HPU_LAZY_MODE=1 python3 run_pipeline.py \ --model_name_or_path microsoft/speecht5_tts \ --text "Hello, my dog is cooler than you!" \ --use_hpu_graphs \ --bf16

@gplutop7 gplutop7 requested a review from regisss as a code owner October 15, 2025 11:59
@HuggingFaceDocBuilderDev
Copy link
Copy Markdown

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

@regisss
Copy link
Copy Markdown
Collaborator

regisss commented Oct 21, 2025

@gplutop7 It seems this test still fails:

def test_text_to_speech(self, model, baseline):

You can run it with

pytest tests/test_pipeline.py -v -s -k "speecht5"

edit: my bad, it runs fine, I had forgotten to add PT_HPU_LAZY_MODE=1

Copy link
Copy Markdown
Collaborator

@regisss regisss left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@regisss regisss merged commit 2fa9dee into huggingface:main Oct 21, 2025
3 of 5 checks passed
gplutop7 pushed a commit to HabanaAI/optimum-habana-fork that referenced this pull request Nov 6, 2025
…=4.55 (huggingface#2308) (huggingface#788)

Co-authored-by: Grzegorz Pluto-Prondzinski <gplutopx@habana.ai>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants