Skip to content

Fix Gemma3IntegrationTest#38471

Merged
ydshieh merged 9 commits intomainfrom
fix_gemma3_ci
May 29, 2025
Merged

Fix Gemma3IntegrationTest#38471
ydshieh merged 9 commits intomainfrom
fix_gemma3_ci

Conversation

@ydshieh
Copy link
Copy Markdown
Collaborator

@ydshieh ydshieh commented May 29, 2025

What does this PR do?

Gemma3IntegrationTest is never actually run. Before #36820, no access to gated repo. Before #38093, @require_read_token didn't work on test class.

All tests are now passing on T4 (with torch 2.7) and A10 (with torch 2.6) - if not skipped by the conditions

@ydshieh ydshieh requested a review from zucchini-nlp May 29, 2025 13:22
@ydshieh
Copy link
Copy Markdown
Collaborator Author

ydshieh commented May 29, 2025

@zucchini-nlp It's just updating the expected values (which are never working before)

EXPECTED_TEXT = EXPECTED_TEXTS.get_expectation()
self.assertEqual(output_text, EXPECTED_TEXT)

@require_torch_large_accelerator
Copy link
Copy Markdown
Collaborator Author

@ydshieh ydshieh May 29, 2025

Choose a reason for hiding this comment

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

The only 4b test that works on T4 is test_model_4b_bf16 (above one). Otherwise are OOM

@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.

Copy link
Copy Markdown
Member

@zucchini-nlp zucchini-nlp left a comment

Choose a reason for hiding this comment

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

Great thanks!

@ydshieh ydshieh merged commit 81cff7a into main May 29, 2025
15 checks passed
@ydshieh ydshieh deleted the fix_gemma3_ci branch May 29, 2025 14:51
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