Skip to content

Use float instead of double to avoid issues with MPS device#499

Merged
younesbelkada merged 1 commit intomainfrom
younesbelkada-patch-1
Jul 6, 2023
Merged

Use float instead of double to avoid issues with MPS device#499
younesbelkada merged 1 commit intomainfrom
younesbelkada-patch-1

Conversation

@younesbelkada
Copy link
Copy Markdown
Contributor

Addresses: #440 (comment)

We should use float instead of double to avoid issues with MPS device

cc @lvwerra

@HuggingFaceDocBuilderDev
Copy link
Copy Markdown

HuggingFaceDocBuilderDev commented Jul 6, 2023

The documentation is not available anymore as the PR was closed or merged.

@younesbelkada younesbelkada requested a review from lvwerra July 6, 2023 08:19
@lvwerra
Copy link
Copy Markdown
Member

lvwerra commented Jul 6, 2023

Can you quickly run the IMDb example to check that we are not breaking anything?

@younesbelkada
Copy link
Copy Markdown
Contributor Author

Sure yes, running it now, will merge after few steps and making sure it converges

@younesbelkada
Copy link
Copy Markdown
Contributor Author

younesbelkada commented Jul 6, 2023

Here are the logs: https://wandb.ai/younesbelkada/trl/runs/zmimbndd?workspace=user- all seems good ! Merging!

@younesbelkada younesbelkada merged commit 64c8db2 into main Jul 6, 2023
@younesbelkada younesbelkada deleted the younesbelkada-patch-1 branch July 6, 2023 08:32
yxliu-TAMU pushed a commit to mincheolseong/ECEN743-GRPO-Project-Proposal that referenced this pull request Apr 20, 2025
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