Skip to content

Make identity/FOOF write covariance dirs, fix target_modules#274

Open
smarter wants to merge 2 commits intoEleutherAI:sourcefrom
smarter:fix-dirs
Open

Make identity/FOOF write covariance dirs, fix target_modules#274
smarter wants to merge 2 commits intoEleutherAI:sourcefrom
smarter:fix-dirs

Conversation

@smarter
Copy link
Copy Markdown
Collaborator

@smarter smarter commented May 5, 2026

This is useful for downstream code in
https://github.com/smarter/memorization_kfac/tree/exps to avoid special cases.

@smarter smarter requested a review from LouisYRYJ May 5, 2026 16:08
Note that I'm not sure PEFT support is fully working in the existing codebase
since we do:
    collector_args = {
        "model": model.base_model,  # type: ignore
        "target_modules": target_modules,
Instead of `"model": model`.
@smarter smarter changed the title Make identity and FOOF write activation/gradient covariance dirs Make identity/FOOF write covariance dirs, fix target_modules May 5, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant