diff --git a/scripts/dev/create_stub_repos.py b/scripts/dev/create_stub_repos.py
index d9722399f..fb6187043 100644
--- a/scripts/dev/create_stub_repos.py
+++ b/scripts/dev/create_stub_repos.py
@@ -96,7 +96,8 @@ def main() -> None:
     )
     touch(
         os.path.join(REPOS, gm, "sgm", "modules", "__init__.py"),
-        "from .conditioner import GeneralConditioner\n",
+        "from .conditioner import GeneralConditioner\n"
+        "from . import attention, diffusionmodules, encoders\n",
     )
     touch(
         os.path.join(REPOS, gm, "sgm", "modules", "diffusionmodules", "openaimodel.py"),