From e2ac501a8db5fca3b03723194b5c5a2e89312552 Mon Sep 17 00:00:00 2001 From: askiiart <90910526+askiiart@users.noreply.github.com> Date: Wed, 21 Dec 2022 15:31:43 -0600 Subject: [PATCH] Fix TypeError bug --- 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(