Closed zhulinJulia24 closed 4 months ago
You may change model = AutoModelForCausalLM.from_pretrained(ckpt_path, torch_dtype=torch.float16, ...
to model = AutoModelForCausalLM.from_pretrained(ckpt_path, torch_dtype=torch.float32, ...
You may change
model = AutoModelForCausalLM.from_pretrained(ckpt_path, torch_dtype=torch.float16, ...
tomodel = AutoModelForCausalLM.from_pretrained(ckpt_path, torch_dtype=torch.float32, ...
model = AutoModelForCausalLM.from_pretrained(ckpt_path, torch_dtype=torch.float32, trust_remote_code=True).cuda() @yuhangzang it works.
Should we change the code in https://huggingface.co/internlm/internlm-xcomposer2-7b ? And the same error occurs by using https://huggingface.co/internlm/internlm-xcomposer2-vl-7b demo code, how to resolve it ?
Thanks for your suggestion, we have modified the readme file of xcomposer2-7b.