From 594498ab844557730ac761d3ee848b71ef6bbbc5 Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Wed, 1 Oct 2025 14:51:25 -0700 Subject: [PATCH] Satisfy ruff --- comfy_extras/nodes_eps.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/comfy_extras/nodes_eps.py b/comfy_extras/nodes_eps.py index 8a90e0caf..c8818f096 100644 --- a/comfy_extras/nodes_eps.py +++ b/comfy_extras/nodes_eps.py @@ -1,11 +1,9 @@ -import torch - 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). - 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" recommended by the paper for its practicality and effectiveness. """ @@ -59,4 +57,4 @@ class EpsilonScaling: NODE_CLASS_MAPPINGS = { "Epsilon Scaling": EpsilonScaling -} \ No newline at end of file +}