pytorch-diffusion
pytorch-diffusion copied to clipboard
Zero noise_rates at all time
https://github.com/quickgrid/pytorch-diffusion/blob/d990fd376c4ac4acd58c410490bc35679c3b436b/ddim.py#L75C20-L75C20
If noise_rate is something between 0-1, wouldn't casting it to long int make it always 0?
Yes, it looks like a mistake in implementation. Need to fix sinusoidal encoding class and unet expecting long integer instead of float.