Add: Add initialization operation of diffusion samples.

chairc · May 6, 2024 · f112797 · f112797
1 parent 3639b6e
commit f112797
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/model/samples/__init__.py b/model/samples/__init__.py
@@ -5,3 +5,7 @@
     @Author : chairc
     @Site   : https://github.com/chairc
 """
+from .base import BaseDiffusion
+from .ddim import DDIMDiffusion
+from .ddpm import DDPMDiffusion
+from .plms import PLMSDiffusion