diff --git a/scripts/dev/create_stub_repos.py b/scripts/dev/create_stub_repos.py
index 2b3711b10..33fd07762 100644
--- a/scripts/dev/create_stub_repos.py
+++ b/scripts/dev/create_stub_repos.py
@@ -46,11 +46,18 @@ def main() -> None:
         os.path.join(REPOS, sd, "ldm", "modules", "attention", "__init__.py"),
         "class CrossAttention:\n    def forward(self, *a, **k): pass\n",
     )
-    touch(os.path.join(REPOS, sd, "ldm", "modules", "diffusionmodules", "__init__.py"))
+    touch(
+        os.path.join(REPOS, sd, "ldm", "modules", "diffusionmodules", "__init__.py"),
+        "from . import model, openaimodel\n",
+    )
     touch(
         os.path.join(REPOS, sd, "ldm", "modules", "diffusionmodules", "model.py"),
         "class AttnBlock:\n    def forward(self, *a, **k): pass\n",
     )
+    touch(
+        os.path.join(REPOS, sd, "ldm", "modules", "diffusionmodules", "openaimodel.py"),
+        "# stub\n",
+    )
     # ldm.modules.midas (sd_models)
     touch(os.path.join(REPOS, sd, "ldm", "modules", "midas", "__init__.py"))
     # ldm.modules.distributions.distributions (textual_inversion.dataset)