diff --git a/model/samples/__init__.py b/model/samples/__init__.py
index 3aa9201..abdd544 100644
--- 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