quality 门控 VAE 快速路径,打通 OpenAI 图像与视频 API
值得精读。核心看点:①用 WeakKeyDictionary 做 nn.Module 级注册表,兼顾对象身份匹配与 GC 友好;②contextmanager 的 previous_enabled 恢复模式让门控可嵌套、异常安全,是请求级开关的干净范式;③_runtime_sampling_quality 把 OpenAI 的 auto 默认值隔离在采样契约之外,是 API 兼容性处理的范例。对要在 diffusion 运行时新增请求级行为开关或维护 OpenAI 兼容入口的开发者有直接参考价值。建议结合 #33818 一起阅读以理解快速路径本体。
参与讨论