jadechoghari commited on
Commit
5309bad
1 Parent(s): 097c338

Update diffusion.py

Browse files
Files changed (1) hide show
  1. diffusion.py +1 -1
diffusion.py CHANGED
@@ -4,7 +4,7 @@
4
  # ADM: https://github.com/openai/guided-diffusion/blob/main/guided_diffusion
5
  # IDDPM: https://github.com/openai/improved-diffusion/blob/main/improved_diffusion/gaussian_diffusion.py
6
 
7
- import gaussian_diffusion as gd
8
  from .respace import SpacedDiffusion, space_timesteps
9
 
10
 
 
4
  # ADM: https://github.com/openai/guided-diffusion/blob/main/guided_diffusion
5
  # IDDPM: https://github.com/openai/improved-diffusion/blob/main/improved_diffusion/gaussian_diffusion.py
6
 
7
+ from . import gaussian_diffusion as gd
8
  from .respace import SpacedDiffusion, space_timesteps
9
 
10