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

Make hf_olmo support AutoModelForCausalLM #631

Merged
merged 6 commits into from
Jun 25, 2024
Merged

Conversation

2015aroras
Copy link
Collaborator

@2015aroras 2015aroras commented Jun 21, 2024

hf_olmo was conflicting with transformers because they were both claiming model_type=olmo. Changing hf_olmo's model type to something else seems to make all our problems go away...

We need to go change all our uploaded models' checkpoints before we merge this though, but thankfully having the wrong model type in config before this gets merged is not a breaking issue (only causes warnings).

@2015aroras 2015aroras requested review from AkshitaB and epwalsh June 22, 2024 00:08
@2015aroras 2015aroras marked this pull request as ready for review June 22, 2024 00:09
Copy link
Member

@epwalsh epwalsh left a comment

Choose a reason for hiding this comment

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

👍

@2015aroras 2015aroras merged commit a5539f4 into main Jun 25, 2024
12 checks passed
@2015aroras 2015aroras deleted the shanea/hf-olmo-auto-model branch June 25, 2024 19:46
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