ImportError: cannot import name 'GaussianDiffusion' from 'diffusion.gaussian_diffusion_ours'

[murmurelo111]

Hello, thank you for your excellent work! I encountered this issue while running bash scripts/val_examples/predict_grab_rndseed_14.sh, and I indeed did not find the GaussianDiffusion class in the diffusion.gaussian_diffusion_ours file. Could you please assist me in resolving this issue? Thank you very much.

[euneestella]

I simply replaced from .gaussian_diffusion_ours import GaussianDiffusion in respace_ours.py to from .gaussian_diffusion import GaussianDiffusion and it worked.