From 9d41d04fde0a0df26bebec3b2ca10db04beeefd9 Mon Sep 17 00:00:00 2001
From: smthemex <138738845+smthemex@users.noreply.github.com>
Date: Mon, 8 Jul 2024 20:55:33 +0800
Subject: [PATCH] Update __init__.py

---
 ldmx/models/diffusion/dpm_solver/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ldmx/models/diffusion/dpm_solver/__init__.py b/ldmx/models/diffusion/dpm_solver/__init__.py
index 8b13789..f56611c 100644
--- a/ldmx/models/diffusion/dpm_solver/__init__.py
+++ b/ldmx/models/diffusion/dpm_solver/__init__.py
@@ -1 +1 @@
-
+from .sampler import DPMSolverSampler