lm-sys/FastChat

Support fastchat-t5-3b-v1.0 on M2 GPU model

オープン

#1,744 opened on 2023/06/20

 (1 件のコメント) (0 件のリアクション) (0 人の担当者)Python (4,736 件のフォーク)batch import
good first issue

Repository metrics

Stars
 (38,959 個のスター)
PR merge metrics
 (PR metrics pending)

説明

CPU model working fine python -m fastchat.serve.cli --model-path /path/to/fastchat-t5-3b-v1.0 --device cpu

GPU models both got errors python -m fastchat.serve.cli --model-path /path/to/fastchat-t5-3b-v1.0 --device mps --load-8bit python -m fastchat.serve.cli --model-path /path/to/fastchat-t5-3b-v1.0 --device mps

So, does fastchat-t5-3b-v1.0 can running on Mac --device mps?

コントリビューターガイド