Fix tiny model script: not using from_pt=True
#27372
Merged
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
What does this PR do?
After #27064, the checkpoint is saved with format
safetensors
. If we specifyfrom_pretrained("...", from_pt=True)
while the checkpoint is insafetensors
format, we get error likeinvalid load key *\xa0* (or *\xd8* etc.)
.This PR remove the usage of
from_pt=True
and therefore fix 1 issue in the failing CI (that checks the script working).Code snippet