KenSchae
/
ComfyUI
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

Not needed anymore.

pull/777/head
comfyanonymous 1 year ago
parent
fb4bf7f591
commit
c9e4a8c9e5
1 changed files with 2 additions and 2 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 4
      comfy/k_diffusion/external.py

4
comfy/k_diffusion/external.py
Unescape View File

@ -134,7 +134,7 @@ class CompVisDenoiser(DiscreteEpsDDPMDenoiser):
"""A wrapper for CompVis diffusion models.""" """A wrapper for CompVis diffusion models."""
def __init__(self, model, quantize=False, device='cpu'): def __init__(self, model, quantize=False, device='cpu'):
super().__init__(model, model.alphas_cumprod.float(), quantize=quantize) super().__init__(model, model.alphas_cumprod, quantize=quantize)
def get_eps(self, *args, **kwargs): def get_eps(self, *args, **kwargs):
return self.inner_model.apply_model(*args, **kwargs) return self.inner_model.apply_model(*args, **kwargs)
@ -173,7 +173,7 @@ class CompVisVDenoiser(DiscreteVDDPMDenoiser):
"""A wrapper for CompVis diffusion models that output v.""" """A wrapper for CompVis diffusion models that output v."""
def __init__(self, model, quantize=False, device='cpu'): def __init__(self, model, quantize=False, device='cpu'):
super().__init__(model, model.alphas_cumprod.float(), quantize=quantize) super().__init__(model, model.alphas_cumprod, quantize=quantize)
def get_v(self, x, t, cond, **kwargs): def get_v(self, x, t, cond, **kwargs):
return self.inner_model.apply_model(x, t, cond) return self.inner_model.apply_model(x, t, cond)

Write Preview
Loading…
Cancel
Save