huggingface/diffusers

Make `FlaxLMSDiscreteScheduler` jittable

Open

#2,180 建立於 2023年1月31日

在 GitHub 查看
 (8 留言) (2 反應) (0 負責人)Python (4,562 fork)batch import
good first issue

倉庫指標

Star
 (22,190 star)
PR 合併指標
 (平均合併 13天 1小時) (30 天內合併 96 個 PR)

描述

Please, see discussion in #2155. I was under the wrong assumption that FlaxLMSDiscreteScheduler had been made jittable as part of #1661, but it never was.

貢獻者指南

研究方向
研究 diffusers 倉庫中 FlaxLMSDiscreteScheduler 的當前實作。查看 issue #2155 中的討論和之前的 PR #1661，以了解之前的嘗試。參考其他已可進行 jit 編譯的 Flax 調度器，以理解其模式。實作該調度器的 jit 編譯，確保所有操作都與 JAX 的 jit 轉換相容。
技術棧
python
領域
performance
議題類型
功能
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
3
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
1-2 天
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
新近可參與
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
清晰
前置要求
Basic knowledge of JAXFamiliarity with diffusers library
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
60