pytorch-diffusion icon indicating copy to clipboard operation
pytorch-diffusion copied to clipboard

Zero noise_rates at all time

Open weizhuowang opened this issue 2 years ago • 1 comments

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?

weizhuowang avatar Jul 05 '23 20:07 weizhuowang

Yes, it looks like a mistake in implementation. Need to fix sinusoidal encoding class and unet expecting long integer instead of float.

quickgrid avatar Jul 06 '23 03:07 quickgrid