FastChat icon indicating copy to clipboard operation
FastChat copied to clipboard

How can I use Multiple NPUs ?

Open QuentinWang1 opened this issue 1 year ago • 1 comments

for example : python3 -m fastchat.serve.cli --model-path lmsys/vicuna-7b-v1.5 --num-gpus 2

maybe this following command need to be supported? python3 -m fastchat.serve.cli --model-path lmsys/vicuna-7b-v1.5 --device npu --num-npus 2 image

image

QuentinWang1 avatar Apr 09 '24 08:04 QuentinWang1

FYI https://github.com/lm-sys/FastChat/issues/3237

wangshuai09 avatar Apr 16 '24 02:04 wangshuai09