diff --git a/hidiffusion/__init__.py b/hidiffusion/__init__.py
new file mode 100644
index 0000000..9c9e6a3
--- /dev/null
+++ b/hidiffusion/__init__.py
@@ -0,0 +1,3 @@
+from .hidiffusion import apply_hidiffusion, remove_hidiffusion
+
+__all__ = ["apply_hidiffusion", "remove_hidiffusion"]