feat: enable save/load after recovery without additional steps

[psyberck]

Description

Extend save and load to also allow saving and loading after recovery without additional steps.

Related Issue

Fixes #603

Type of Change

• Bug fix (non-breaking change which fixes an issue)
• New feature (non-breaking change which adds functionality)
• Refactor (no functional change)
• Breaking change (fix or feature that would cause existing functionality to not work as expected)
• Documentation update

Testing

• I added or updated tests covering my changes
• Existing tests pass locally (uv run pytest -m "cpu and not slow")

For full setup and testing instructions, see the Contributing Guide.

Checklist

• My code follows the style guidelines of this project
• I have performed a self-review of my code, especially for agent-assisted changes
• I updated the documentation where necessary

---

Thanks for contributing to Pruna! We're excited to review your work.

New to contributing? Check out our Contributing Guide for everything you need to get started.

Note:

• Draft PRs or PRs without a clear and detailed overview may be delayed.
• Please mark your PR as Ready for Review and ensure the sections above are filled out.
• Contributions that are entirely AI-generated without meaningful human review are discouraged.

---

First Prune (1-year OSS anniversary)

First Prune marks one year of Pruna’s open-source work. During the initiative window, qualifying merged contributions count toward First Prune. You can earn credits for our performance models via our API.

If you’d like your contribution to count toward First Prune, here’s how it works:

• Initiative window: First Prune starts on March 31.
• Issue assignment: For your PR to count toward First Prune, the related issue must be assigned to the contributor opening the PR. Issues are labeled with first-prune.
• Open for review: Please open your PR and mark it ready for review by April 30 (end of April).
• Review priority: We’ll make our best effort to review quickly any PR that is open and has a review request before April 30.
• Credits: Each qualifying merged PR earns 30 credits. We’ll be in touch after all qualifying PRs for First Prune have been merged.
• To get started: Have a look at all models. You’ll need to sign up on the dashboard before you can redeem your credits.

[codacy-production]

Not up to standards ⛔

🔴 Issues 3 high

Alerts:
⚠ 3 issues (≤ 0 issues of at least minor severity)

Results:
3 new issues

Category	Results
Security	3 high

View in Codacy

🟢 Metrics 12 complexity · 0 duplication

Metric	Results
Complexity	12
Duplication	0

View in Codacy

_{TIP This summary will be updated as you push new changes. Give us feedback}

[sdiazlor]

Hi @psyberck! Thank you for your contribution! I've run the tests. Please take a look before we jump into any review :)

[psyberck]

Hey @sdiazlor , can you help kindly re-run the tests? There seems some network failures during the tests:

ERROR tests/algorithms/test_algorithms.py::test_full_integration[TestIFW_whisper_tiny_random-cpu] - huggingface_hub.errors.LocalEntryNotFoundError: An error happened while trying to locate the files on the Hub and we cannot find the appropriate snapshot folder for the specified revision on the local disk. Please check your internet connection and try again.

[sdiazlor]

Sure thing! I ran them again :)

[github-actions]

This PR has been inactive for 10 days and is now marked as stale.

[simlang]

Thanks for looking at this! I left a view requests :)

[sdiazlor]

Hi @psyberck! When you have time, could you take a look at the comments? :)

[psyberck]

Thanks for the comments @simlang. I will update accordingly tonight @sdiazlor :)

[psyberck]

Code updated @simlang cc @sdiazlor

[github-actions]

This PR has been inactive for 10 days and is now marked as stale.

[cursor]

Cursor Bugbot has reviewed your changes and found 1 potential issue.

^{❌ Bugbot Autofix is OFF. To automatically fix reported issues with cloud agents, enable autofix in the Cursor dashboard.}

Comment @cursor review or bugbot run to trigger another review on this PR

^{Reviewed by Cursor Bugbot for commit 81ca7d0. Configure here.}

[cursor]

Hook saves pre-apply model

High Severity

post_apply_hook receives the model from before _apply, but PERP recovery can return a different object (e.g. after finetune unwrap or LoRA base_model). refresh_saved_model then rewrites the save-before-smash cache from that stale reference, so save/load can miss recovered weights.

Additional Locations (1)

• src/pruna/algorithms/global_utils/recovery/perp_recoverer.py#L199-L201

 

^{Reviewed by Cursor Bugbot for commit 81ca7d0. Configure here.}