From f5365c9c815318169a584ccde763df37f25f356c Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Sun, 26 Mar 2023 00:36:54 -0400 Subject: [PATCH] Fix ddim for Mac: #264 --- comfy/ldm/models/diffusion/ddim.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/ldm/models/diffusion/ddim.py b/comfy/ldm/models/diffusion/ddim.py index fe39c76c7..5e2d73645 100644 --- a/comfy/ldm/models/diffusion/ddim.py +++ b/comfy/ldm/models/diffusion/ddim.py @@ -18,7 +18,7 @@ class DDIMSampler(object): def register_buffer(self, name, attr): if type(attr) == torch.Tensor: if attr.device != self.device: - attr = attr.to(self.device) + attr = attr.float().to(self.device) setattr(self, name, attr) def make_schedule(self, ddim_num_steps, ddim_discretize="uniform", ddim_eta=0., verbose=True):