[FEATURE] Implement save/load model after recovering #603
Description
✨ What You’ll Do
During our latest release, we have included the recovery algorithms as a new algorithm family. Extend save and load to also allow saving and loading after recovery without additional steps.
# Save the model
smashed_model.save_pretrained("saved_recovered_model")
#Load the model
loaded_model = PrunaModel.from_pretrained("saved_recovered_model/")
🤖 Useful Resources
- The pruna documentation: https://docs.pruna.ai/
- Save/Load documentation: https://docs.pruna.ai/en/stable/docs_pruna/user_manual/save_load.html
- Check how recovery works under /algoritms and check the list here: https://docs.pruna.ai/en/stable/compression.html#id1
✅ Acceptance Criteria
- It follows the style guidelines.
- Tests demonstrate that the recovered model is correctly saved and updated.
- The metric integrates properly into Pruna’s evaluation framework.
And don’t forget to give us a ⭐️!
❓ Questions?
Feel free to jump into the #contributing Discord channel if you hit any roadblocks. Can’t wait to see your contribution! 🚀
