xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-14 08:16:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

Satisfy ruff

Browse Source
This commit is contained in:
Jedrzej Kosinski 2025-10-01 14:51:25 -07:00
parent 9a53594029
commit 594498ab84
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy_extras/nodes_eps.py
View File

@ -1,11 +1,9 @@
import torch
class EpsilonScaling: class EpsilonScaling:
""" """
Implements the Epsilon Scaling method from 'Elucidating the Exposure Bias in Diffusion Models' Implements the Epsilon Scaling method from 'Elucidating the Exposure Bias in Diffusion Models'
(https://arxiv.org/abs/2308.15321v6). (https://arxiv.org/abs/2308.15321v6).
This method mitigates exposure bias by scaling the predicted noise during sampling, This method mitigates exposure bias by scaling the predicted noise during sampling,
which can significantly improve sample quality. This implementation uses the "uniform schedule" which can significantly improve sample quality. This implementation uses the "uniform schedule"
recommended by the paper for its practicality and effectiveness. recommended by the paper for its practicality and effectiveness.
""" """
@ -59,4 +57,4 @@ class EpsilonScaling:
NODE_CLASS_MAPPINGS = { NODE_CLASS_MAPPINGS = {
"Epsilon Scaling": EpsilonScaling "Epsilon Scaling": EpsilonScaling
} }