Skip to content

Bnb test fix for different hardwares#29066

Merged
amyeroberts merged 3 commits into
huggingface:mainfrom
Titus-von-Koeller:bnb-test-fix-for-different-hardwares
Feb 19, 2024
Merged

Bnb test fix for different hardwares#29066
amyeroberts merged 3 commits into
huggingface:mainfrom
Titus-von-Koeller:bnb-test-fix-for-different-hardwares

Conversation

@Titus-von-Koeller

Copy link
Copy Markdown
Contributor

Just updating the acceptable generated text, as this is slightly different based on hardware from what I can tell. The first commit is based on what I observed on my dev VM with A10G and the second commit is based on what I saw in the failing BNB integration pipeline.

In my eyes, both text strings seem similar enough. I already spoke about this with @younesbelkada regarding the first commit. The second seems a similar case.

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

@younesbelkada younesbelkada left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

Thank you very much for updating the tests so that they pass on a A10 !

Comment thread tests/quantization/bnb/test_mixed_int8.py
Comment thread tests/quantization/bnb/test_mixed_int8.py
add explanatory comments

Co-authored-by: Younes Belkada <[email protected]>

@amyeroberts amyeroberts left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

LGTM - thanks for updating!

@Titus-von-Koeller

Copy link
Copy Markdown
Contributor Author

Could one of you please merge? I don't have permission to do so.

@amyeroberts
amyeroberts merged commit 5ce90f3 into huggingface:main Feb 19, 2024
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.

4 participants