Update trackio integration to use Buckets and "freeze" Space after training - #45329
Conversation
trackio integration to use Buckets and "freeze" Space after training
|
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update. |
SunMarc
left a comment
There was a problem hiding this comment.
Thanks ! Left some minor comments
| gradio_space_id = self._resolve_trackio_space_id(args) | ||
| if gradio_space_id is None: | ||
| return |
There was a problem hiding this comment.
from the code in _resolve_trackio_space_id, it feels like gradio_space_id is never None no or SQLiteStorage.get_space_id(args.project) can return None ? Also maybe we can rename a bit the method to call it get_trackio_space_id maybe ?
| @@ -1041,13 +1141,22 @@ def on_push_begin(self, args, state, control, model, **kwargs): | |||
|
Thanks again for the thorough review @SunMarc! I think I've addressed everything (cc @qgallouedec for visibility) Edit: I think this broke the links on the model card, let me fix that |
SunMarc
left a comment
There was a problem hiding this comment.
Thanks a lot for iterating ! Can you have a look @qgallouedec ?
|
@bot /style |
|
Style fix bot fixed some files and pushed the changes. |
|
Should be all set now! Tested end-to-end with this model: https://huggingface.co/abidlabs/trackio-transformers-demo-493111 and the Static Space shows up as expected: Screen.Recording.2026-04-10.at.11.19.19.AM.mov |
…training (huggingface#45329) * changes * changes * changes * changes * changes * changes * changes * changes * changes * changes * changes * changes * changes * changes * changes * changes * changes * changes * changes * changes * changes * changes * changes * changes * changes * changes * changes * changes * changes * changes * changes * changes * changes * changes * Apply repo consistency fixes * changes * changes * changes * changes * changes * changes * changes * changes * chore: empty commit --------- Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
Updates
TrackioCallbackandTrainingArgumentsfor the latest version of Trackio using HF Buckets as the backend, and control over creating a static Space for the Trackio dashboard during or at the end of training.These are now the
TrainingArgumentsrelevant fortrackio:trackio_space_id— Only affects the live read and write Gradio Space used during training.None(default): metrics stay local (no Gradio Space for logging).str: create/use that Gradio Space and stream metrics there for the run.trackio_static_space_id— Controls the creation of a static read-only Trackio Space:False: no static Space — nosync(sdk="static")on Hub push, nofreezeafter training.None(default) orstr: a static Trackio Space is created in two cases:trackio.sync(..., sdk="static")so the model card can link a static dashboard; auto Space name ifNone, or use the string as the static Space id.trackio_space_idwas set (Gradio):trackio.freezefrom that Gradio Space;Nonelets Trackio name the static Space,strsets the destination id.Example of static Space created: https://huggingface.co/spaces/abidlabs/trackio-transformers-demo-493111-static-d0a1cc
Example of model card created: https://huggingface.co/abidlabs/trackio-transformers-demo-493111