Remove unused default reference

Remove unused default reference
This commit is contained in:
John Slegers 2024-08-12 03:25:33 +02:00
parent d078cb1683
commit 017167c3bb

View File

@ -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