Open LittlefishStudent opened 1 year ago
加载YeungNLP/firefly-llama2-13b时报如上错误
已解决!我是合并模型后,出现这个报错,在AutoModelForCausalLM.from_pretrained中加入参数load_in_4bit=True或者load_in_8bit=True就解决了。