From e504e0e5545cef2aed9e02e51b6c13940fcb8290 Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Tue, 12 Apr 2022 13:02:18 -0700 Subject: [PATCH] cleanup --- denoising_diffusion_pytorch/denoising_diffusion_pytorch.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py b/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py index 1153ecd..bcc4166 100644 --- a/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py +++ b/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py @@ -335,8 +335,6 @@ class GaussianDiffusion(nn.Module): self.num_timesteps = int(timesteps) self.loss_type = loss_type - to_torch = partial(torch.tensor, dtype=torch.float32) - self.register_buffer('betas', betas) self.register_buffer('alphas_cumprod', alphas_cumprod) self.register_buffer('alphas_cumprod_prev', alphas_cumprod_prev)