Skip to content

why previously updated knowledge covariance is not added in AlphaEdit implementation? #8

@jasonrichdarmawan

Description

@jasonrichdarmawan

upd_matrix = torch.linalg.solve(
P[i,:,:].cuda() @ (layer_ks @ layer_ks.T) + hparams.L2*torch.eye(layer_ks.shape[0], dtype=torch.float,device="cuda"), P[i,:,:].cuda() @ layer_ks @ resid
)

This is different to the original AlphaEdit implementation

https://github.com/jianghoucheng/AlphaEdit/blob/75e117bcbbc58b1fd28df41e1cbb2fa3d25e3ef1/AlphaEdit/AlphaEdit_main.py#L130-L132

Image

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions