[FEATURE] Implement save/load model after recovering

[sdiazlor]

‼️ If you want to work on this issue: please comment below and wait until a maintainer assigns this issue to you before opening a PR to avoid several contributions on the same issue. Thanks! 😊

---

✨ 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! 🚀

---

Share on Socials

[psyberck]

Hey @sdiazlor , if this issue is still not assigned, I will be happy to work on it. Thanks.

[sdiazlor]

Hi @psyberck, sure thing. Thanks for participating!