diff --git a/litellm/images/__init__.py b/litellm/images/__init__.py
new file mode 100644
index 00000000000..e69de29bb2d
diff --git a/litellm/proxy/image_endpoints/__init__.py b/litellm/proxy/image_endpoints/__init__.py
new file mode 100644
index 00000000000..2e24ea01e5a
--- /dev/null
+++ b/litellm/proxy/image_endpoints/__init__.py
@@ -0,0 +1,3 @@
+from .endpoints import image_generation
+
+__all__ = ["image_generation"]
diff --git a/tests/proxy_unit_tests/test_key_generate_prisma.py b/tests/proxy_unit_tests/test_key_generate_prisma.py
index cff4747bc0e..e1890732ec0 100644
--- a/tests/proxy_unit_tests/test_key_generate_prisma.py
+++ b/tests/proxy_unit_tests/test_key_generate_prisma.py
@@ -77,6 +77,7 @@ from litellm.proxy.proxy_server import (
     moderations,
     user_api_key_auth,
 )
+from litellm.proxy.image_endpoints import image_generation
 from litellm.proxy.management_endpoints.customer_endpoints import (
     new_end_user,
 )