Skip to content

Conversation

@norabelrose
Copy link
Collaborator

Adds MistralModel to the model surgery functions

@norabelrose norabelrose requested a review from levmckinney May 23, 2024 20:10
@norabelrose
Copy link
Collaborator Author

Some very strange test failure is happening involving datapipes

@levmckinney
Copy link
Collaborator

levmckinney commented May 30, 2024

Looks like their is an issue with pytorch 2.3 that is causing it to be incompatible with torchdata pytorch/pytorch#126094.

It should be fixed in pytorch 2.3.1

@levmckinney
Copy link
Collaborator

@norabelrose rebase or merge this with main and it should work.

Copy link
Collaborator

@levmckinney levmckinney left a comment

Choose a reason for hiding this comment

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

LGTM

@norabelrose norabelrose merged commit abdac0c into AlignmentResearch:main Aug 7, 2025
1 of 6 checks passed
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.

2 participants