Как я могу реализовать собственный планировщик Эйлера-Маруямы в диффузорах Hugging Face для модели FLUX? [закрыто] ⇐ Python

[Anonymous]

Я работаю с моделью FLUX.1-dev от Hugging Face, используя библиотеку диффузоров и пытаюсь реализовать собственный планировщик на основе метода Эйлера-Маруямы (SDE). Моя цель — упростить конвейер генерации изображений и заменить планировщик по умолчанию своей собственной реализацией.
Я понимаю общую идею Эйлера-Маруямы для стохастических дифференциальных уравнений, но не уверен, как правильно интегрировать ее в структуру диффузоров, чтобы она правильно работала с циклом шумоподавления модели. В частности, я не уверен в следующем:
Как структурировать класс планировщика, чтобы он был совместим с API-интерфейсами диффузоров.
Как правильно обновлять скрытые значения на каждом временном шаге, используя условия дрейфа и шума.
Как сопоставить временные шаги и масштабы шума с ожидаемыми входными данными модели.
Я пытался адаптировать существующий планировщик в качестве эталона, но не уверен, что мое обновление шаг является математически или структурно правильным.
Как правильно реализовать и подключить собственный планировщик Эйлера-Маруямы для модели диффузии, такой как FLUX, в диффузорах?

Подробнее здесь: https://stackoverflow.com/questions/798 ... -diffusers