fix: imports bug

wandb · Oct 8, 2024 · 376d3ab · 376d3ab
1 parent 185669a
commit 376d3ab
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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"]
+