diff --git a/hemm/eval_pipelines/__init__.py b/hemm/eval_pipelines/__init__.py
index e95cb96..ae451df 100644
--- a/hemm/eval_pipelines/__init__.py
+++ b/hemm/eval_pipelines/__init__.py
@@ -1,4 +1,4 @@
 from .eval_pipeline import EvaluationPipeline
-from .model import BaseDiffusionModel
 
-__all__ = ["BaseDiffusionModel", "EvaluationPipeline"]
+__all__ = ["EvaluationPipeline"]
+