two mistakes

[YMKiii]

1.dataloader_train.py line 72-75，missed T.ToTensor(), which is include /255. T.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)), only ensure -0.5 and /0.5. recommend use PIL to preprocess image instead of use opencv.
2. diffusion.py line 79-83, def add_noise_to_img(self, img, t): return (sqrt_alpha_timestep * epsilon) + (sqrt_one_minus_alpha_timestep * epsilon), epsilon, which is should be return (sqrt_alpha_timestep * img) + (sqrt_one_minus_alpha_timestep * epsilon), epsilon.