From f1127979f9ad1101f7aa9e43d18c605bd3967701 Mon Sep 17 00:00:00 2001
From: ChairC <974833488@qq.com>
Date: Mon, 6 May 2024 11:08:21 +0800
Subject: [PATCH] Add: Add initialization operation of diffusion samples.

---
 model/samples/__init__.py | 4 ++++
 1 file changed, 4 insertions(+)

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