search

mymusise / ChatGLM-Tuning

基于ChatGLM-6B + LoRA的Fintune方案
MIT License
3.73k stars 440 forks source link

AttributeError: 'NoneType' object has no attribute 'device' #164

Open bolongliu opened 1 year ago

bolongliu commented 1 year ago

将 model = PeftModel.from_pretrained(model, "./output/") 改为： model = PeftModel.from_pretrained(model, "./output/", device_map={'':0})