Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix device of masks in tests #27887

Merged
merged 1 commit into from
Dec 7, 2023
Merged

Conversation

fxmarty
Copy link
Contributor

@fxmarty fxmarty commented Dec 7, 2023

As per title, fix a bug introduced in #24587 in the GPU tests.

@fxmarty fxmarty requested review from ArthurZucker and LysandreJik and removed request for ArthurZucker December 7, 2023 11:47
@fxmarty fxmarty assigned amyeroberts and unassigned amyeroberts Dec 7, 2023
Copy link
Collaborator

@ArthurZucker ArthurZucker left a comment

Choose a reason for hiding this comment

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

Thanks

@ArthurZucker ArthurZucker removed the request for review from amyeroberts December 7, 2023 11:53
@ydshieh
Copy link
Collaborator

ydshieh commented Dec 7, 2023

Strange that we don't see the issue on our CI. Do you know why @fxmarty?

@fxmarty
Copy link
Contributor Author

fxmarty commented Dec 7, 2023

@ydshieh Can you try RUN_SLOW=1 CUDA_VISIBLE_DEVICES=0 pytest tests/models/llama -s -vvvvv -k "test_resize_tokens_embeddings" on main?

@fxmarty
Copy link
Contributor Author

fxmarty commented Dec 7, 2023

I am not sure it's weird. The push tests should run on GPU so those should have failed...

machine_type: [single-gpu]

@fxmarty
Copy link
Contributor Author

fxmarty commented Dec 7, 2023

As discussed offline it did actually fail (https://github.com/huggingface/transformers/actions/runs/7124821720) so all good!

@fxmarty fxmarty merged commit c99f254 into huggingface:main Dec 7, 2023
18 checks passed
nevikw39 pushed a commit to NTHU-ML-2023-team19/transformers that referenced this pull request Dec 7, 2023
nevikw39 pushed a commit to NTHU-ML-2023-team19/transformers that referenced this pull request Dec 7, 2023
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