autotrain-advanced icon indicating copy to clipboard operation
autotrain-advanced copied to clipboard

Allow pointing to a revision for models

Open aflah02 opened this issue 4 months ago • 0 comments

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

aflah02 avatar Oct 09 '24 14:10 aflah02