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 合并指标
 (平均合并 14天 10小时) (30 天内合并 101 个 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.

贡献者指南

研究方向
调查FlaxLMSDiscreteScheduler的实现，找出与JAX的JIT编译不兼容的操作（如动态控制流或非静态形状），并重构使其可jit。
技术栈
python
领域
machine learningai
议题类型
缺陷
难度面向新贡献者的预计实现难度，1 表示很小改动，5 表示专家级工作。
3
预计时间有经验贡献者完成调查、实现、测试并准备 pull request 的粗略时间范围。
半天
活动状态议题当前的可参与程度：新鲜、活跃、陈旧、阻塞或等待维护者输入。
活跃
清晰度议题是否清楚说明期望改动、验收标准和下一步。
清晰
前置要求
PythonJAX
新手友好度1-100 的估计分数，表示该议题对首次贡献者的友好程度。
40