search

WangRongsheng / XrayGLM

🩺 首个会看胸部X光片的中文多模态医学大模型 | The first Chinese Medical Multimodal Model that Chest Radiographs Summarization.
Other
884 stars 128 forks source link

为什么我用--qlora训练以后的权重，不能在cli_demo.py中正常加载？ #37

Closed Xjmengnieer closed 1 year ago

Xjmengnieer commented 1 year ago

image 显示是输入数据尺寸不对，我打印了一下这里的self.wigth.data,其shape为： image 而后，我再打印了一下这里加载的权重：state_dict[prefix+'weight']，其shape为： image 这个值恰好是上面self.wigth.data的shape的乘积的一半，而且上面self.wigth.data是float16， 而state_dict[prefix+'weight']是uint8， 能给一点意见吗？感谢

boh95 commented 1 year ago

遇到了同样的问题。请问这个问题有解决吗？ @Xjmengnieer