diff --git a/comfy/ldm/modules/diffusionmodules/upscaling.py b/comfy/ldm/modules/diffusionmodules/upscaling.py index f5ac7c2f9..972cc76f0 100644 --- a/comfy/ldm/modules/diffusionmodules/upscaling.py +++ b/comfy/ldm/modules/diffusionmodules/upscaling.py @@ -4,7 +4,6 @@ import numpy as np from functools import partial from .util import extract_into_tensor, make_beta_schedule -from comfy.ldm.util import default class AbstractLowScaleModel(nn.Module): @@ -80,6 +79,3 @@ class ImageConcatWithNoiseAugmentation(AbstractLowScaleModel): assert isinstance(noise_level, torch.Tensor) z = self.q_sample(x, noise_level, seed=seed) return z, noise_level - - -