From b3462ac140c1016bd0e563850219f078cd11e5d6 Mon Sep 17 00:00:00 2001 From: Sean Lavery Date: Wed, 21 Dec 2022 12:25:59 -0800 Subject: [PATCH] fix runtime error on windows changed return from float to int to prevent error. This fixes issue 6 --- point_e/diffusion/k_diffusion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/point_e/diffusion/k_diffusion.py b/point_e/diffusion/k_diffusion.py index 385486d..35f642d 100644 --- a/point_e/diffusion/k_diffusion.py +++ b/point_e/diffusion/k_diffusion.py @@ -93,7 +93,7 @@ def sigma_to_t(self, sigma): elif alpha_cumprod <= self.diffusion.alphas_cumprod[-1]: return self.diffusion.num_timesteps - 1 else: - return float(self.alpha_cumprod_to_t(alpha_cumprod)) + return int(self.alpha_cumprod_to_t(alpha_cumprod)) def denoise(self, x_t, sigmas, clip_denoised=True, model_kwargs=None): t = th.tensor(