From e6e50ab2dd1ff02a8ffd4e6954498771ef1c4ad2 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Fri, 16 Jun 2023 17:16:51 -0400 Subject: [PATCH] Fix an issue when alphas_comprod are half floats. --- 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 49ce5ae3..d15eb595 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, quantize=quantize) + super().__init__(model, model.alphas_cumprod.float(), 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, quantize=quantize) + super().__init__(model, model.alphas_cumprod.float(), quantize=quantize) def get_v(self, x, t, cond, **kwargs): return self.inner_model.apply_model(x, t, cond)