foundation-model-stack / fms-hf-tuning

🚀 Collection of tuning recipes with HuggingFace SFTTrainer and PyTorch FSDP.
Apache License 2.0
22 stars 41 forks source link

bug: Missing on_save() event in callback.py #178

Closed seshapad closed 2 months ago

seshapad commented 3 months ago

Describe the bug

There is no callback available for on_save event in the trainer controller callback.

Platform

Please provide details about the environment you are using, including the following:

Sample Code

NA

Expected behavior

Any operation registered with on_save trigger or metric registered with it should work.

Observed behavior

on_save is not available in callback.

Additional context

Add any other context about the problem here.