autotrain-advanced
autotrain-advanced copied to clipboard
Allow pointing to a revision for models
Feature Request
Several models like Pythia, OLMo etc. use revisions to store different checkpoints. It would be great if supplying a revision arg is supported for these models.
Motivation
Just to allow people to point to intermediate checkpoints instead of only the final checkpoint
Additional Context
A similar issue + PR exists for datasets - https://github.com/huggingface/autotrain-advanced/issues/740 but there has been no communication on that PR and hence I wanted to ask if such a feature will be accepted before raising a PR