Skip to content

Conversation

@philschmid
Copy link
Contributor

@philschmid philschmid commented Nov 3, 2023

This PR adds the revision key to the push_to_hub method to push models, to specific revision, this is helpful ,e.g. to version compiled checkpoints, e.g. 2.15.0 revision.

Also removes/fixes fairscale change huggingface/transformers#25702

@HuggingFaceDocBuilderDev

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint.

@dacorvo
Copy link
Collaborator

dacorvo commented Nov 6, 2023

Thanks for this pull-request.
Regarding the addition of revision, I think it was omitted because the original method in optimum does not have it.
For consistency, it would be nice to add it also here:

.

@philschmid
Copy link
Contributor Author

@dacorvo added it for the decoder as well.

@philschmid philschmid merged commit 8abbcbd into main Nov 6, 2023
@philschmid philschmid deleted the add-revision branch November 6, 2023 16:21
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.

6 participants