diff --git a/stable_diffusion_tf/__init__.py b/stable_diffusion_tf/__init__.py
index e69de29..7889f82 100644
--- a/stable_diffusion_tf/__init__.py
+++ b/stable_diffusion_tf/__init__.py
@@ -0,0 +1 @@
+from stable_diffusion_tf.stable_diffusion import StableDiffusion