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