File "E:\minimal-diffusion-main\main_single.py", line 38, in
/ unsqueeze3x(scalars.alpha_bar[t].sqrt())
IndexError: tensors used as indices must be long, byte or bool tensors
self.get_x0_from_xt_eps = lambda xt, eps, t, scalars: (
self.clamp_x0(
1
/ unsqueeze3x(scalars.alpha_bar[t].sqrt())
* (xt - unsqueeze3x((1 - scalars.alpha_bar[t]).sqrt()) * eps)
)
)
File "E:\minimal-diffusion-main\main_single.py", line 38, in
/ unsqueeze3x(scalars.alpha_bar[t].sqrt())
IndexError: tensors used as indices must be long, byte or bool tensors
self.get_x0_from_xt_eps = lambda xt, eps, t, scalars: (
self.clamp_x0(
1
/ unsqueeze3x(scalars.alpha_bar[t].sqrt())
* (xt - unsqueeze3x((1 - scalars.alpha_bar[t]).sqrt()) * eps)
)
)