fastllm icon indicating copy to clipboard operation
fastllm copied to clipboard

baichuan2flm.py ..python3.10/site-packages/accelerate/big_modeling.py", line 415, in wrapper raise RuntimeError("You can't move a model that has some modules offloaded to cpu or disk.") RuntimeError: You can't move a model that has some modules offloaded to cpu or disk.

Open zxzxde opened this issue 1 year ago • 2 comments

调用的是baichuan2flm.py转的是Baichuan2-7B-Chat模型时报错。 /fastllm/build/tools/baichuan2flm.py", line 12, in model.to("cpu")

python3.10/site-packages/accelerate/big_modeling.py", line 415, in wrapper raise RuntimeError("You can't move a model that has some modules offloaded to cpu or disk.") RuntimeError: You can't move a model that has some modules offloaded to cpu or disk.

zxzxde avatar Sep 30 '23 09:09 zxzxde

不要放到cpu上,指定放到gpu卡上 export CUDA_VISIBLE_DEVICES='0' @zxzxde

wut0n9 avatar Oct 09 '23 07:10 wut0n9

你好如果我没有gpu呢该如何做呢@wut0n9

hariji814 avatar Dec 11 '23 06:12 hariji814