lm-sys/FastChat

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

開放

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

在 GitHub 查看
 (1 則留言) (0 個反應) (0 位負責人)Python (4,736 個分叉)batch import
good first issue

倉庫指標

星標
 (38,959 顆星)
PR 合併指標
 (PR 指標待抓取)

描述

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?

貢獻者指南

研究方向
該問題報告 fastchat t5-3b v1.0 在 MPS GPU 上失敗，而 CPU 工作正常。錯誤可能出現在模型載入或裝置映射程式碼中。請研究 fastchat.serve.cli 模組及其如何處理 device 和 load 8bit 旗標。該模型可能需要特定的量化或 MPS 特定調整。請在 FastChat 儲存庫中查找類似問題或修補程式。使用者未提供完整的錯誤日誌，因此需要使用給定的命令重現錯誤。
技術棧
pythonpytorch
領域
backendmachine learning
議題類型
錯誤
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
2
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
1-3 小時
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
活躍
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
清晰
前置要求
PythonPyTorchGit
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
70