MEMIT不支持chatglm吗？

zjunlp / EasyEdit

[ACL 2024] An Easy-to-use Knowledge Editing Framework for LLMs.

https://zjunlp.github.io/project/KnowEdit

MIT License

1.74k stars 210 forks source link

Closed bebr2 closed 3 months ago

bebr2 commented 3 months ago

作者您好！我在使用MEMIT修改chatglm时报错：

 target_init = cur_out[0][0, lookup_idxs[0]].detach().clone()
IndexError: index 7 is out of bounds for dimension 1 with size 7

搜索之前的issue似乎也有类似问题，请问这个问题现在是还没有解决吗？谢谢！

zxlzr commented 3 months ago

您好，前段时间算力紧张这部分临时搁置了，实在没卡了，我们会尽快优化修复这个问题。

bebr2 commented 3 months ago

理解，作者大大辛苦！