From c9e4a8c9e5a5baabd86fdb192ef79d3b288408a6 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Sun, 18 Jun 2023 13:06:59 -0400 Subject: [PATCH] Not needed anymore. --- comfy/k_diffusion/external.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/comfy/k_diffusion/external.py b/comfy/k_diffusion/external.py index d15eb595..49ce5ae3 100644 --- a/comfy/k_diffusion/external.py +++ b/comfy/k_diffusion/external.py @@ -134,7 +134,7 @@ class CompVisDenoiser(DiscreteEpsDDPMDenoiser): """A wrapper for CompVis diffusion models.""" 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): return self.inner_model.apply_model(*args, **kwargs) @@ -173,7 +173,7 @@ class CompVisVDenoiser(DiscreteVDDPMDenoiser): """A wrapper for CompVis diffusion models that output v.""" 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): return self.inner_model.apply_model(x, t, cond)