Skip to content

[testing] update test_longcat_generation_cpu#41368

Merged
ydshieh merged 2 commits intomainfrom
fix_longcat
Oct 6, 2025
Merged

[testing] update test_longcat_generation_cpu#41368
ydshieh merged 2 commits intomainfrom
fix_longcat

Conversation

@ydshieh
Copy link
Copy Markdown
Collaborator

@ydshieh ydshieh commented Oct 6, 2025

What does this PR do?

Use "auto" instead of "cpu" to avoid the pytest process being killed.

(#40132 has issue with using "auto" for this model, and we still need an fix for that)

Also using max_new_tokens=3 to run this test within 8 minutes (on a single A10 CI runner)

The expected output is not updated yet: will do it after the above mentioned issue is fixed.

@ydshieh ydshieh requested a review from molbap October 6, 2025 12:51
@github-actions
Copy link
Copy Markdown
Contributor

github-actions Bot commented Oct 6, 2025

[For maintainers] Suggested jobs to run (before merge)

run-slow: longcat_flash

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

@molbap molbap left a comment

Choose a reason for hiding this comment

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

Just a change needed for the test to work, LGTM otherwise!

Comment thread tests/models/longcat_flash/test_modeling_longcat_flash.py Outdated
Co-authored-by: Pablo Montalvo <39954772+molbap@users.noreply.github.com>
@ydshieh ydshieh enabled auto-merge (squash) October 6, 2025 13:14
@ydshieh ydshieh merged commit 6bf6e36 into main Oct 6, 2025
20 checks passed
@ydshieh ydshieh deleted the fix_longcat branch October 6, 2025 13:21
AhnJoonSung pushed a commit to AhnJoonSung/transformers that referenced this pull request Oct 12, 2025
* fix

* Update tests/models/longcat_flash/test_modeling_longcat_flash.py

Co-authored-by: Pablo Montalvo <39954772+molbap@users.noreply.github.com>

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
Co-authored-by: Pablo Montalvo <39954772+molbap@users.noreply.github.com>
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