huggingface / autotrain-advanced

🤗 AutoTrain Advanced
https://huggingface.co/autotrain
Apache License 2.0
4.13k stars 503 forks source link

Allow pointing to a revision for models #788

Open aflah02 opened 1 month ago

aflah02 commented 1 month ago

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

github-actions[bot] commented 3 weeks ago

This issue is stale because it has been open for 30 days with no activity.

aflah02 commented 3 weeks ago

@abhishekkrthakur Any thoughts?