lm-sys/FastChat

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

Open

#1,744 建立於 2023年6月20日

在 GitHub 查看
 (1 留言) (0 反應) (0 負責人)Python (38,959 star) (4,736 fork)batch import
good first issue

描述

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?

貢獻者指南

技術棧
python
領域
machine learning
議題類型
bug
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
3
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
half day
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
needs maintainer response
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
clear
前置要求
Basic knowledge of FastChatMPS device setup
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
30
研究方向
The issue reports that fastchat t5-3b v1.0 fails on MPS GPU while CPU works. The error is likely in the model loading or device mapping code. Investigate the fastchat.serve.cli module and how it handles the device and load 8bit flags. The model may need specific quantization or MPS specific adjustments. Look for similar issues or patches in the FastChat repository. The user did not provide full error logs, so reproducing the error with the given commands is necessary.