Missing Code for MEMIT_prune

jianghoucheng / AlphaEdit

36 stars 4 forks source link

Missing Code for MEMIT_prune #6

Closed xpq-tech closed 3 hours ago

xpq-tech commented 4 hours ago

Hello,

AlphaEdit is a fantastic work!

I noticed that in evaluate.py, both MEMIT and MEMIT_prune refer to apply_memit_to_model. This seems like it might be an error. Could you look into this?

Thank you!

xpq-tech commented 3 hours ago

I found that MEMIT_prune is indeed executed in evaluate.py. Thanks!