From 80b305be36c5eb310fc93f678839f7938f3c311a Mon Sep 17 00:00:00 2001 From: Michael Cahill Date: Sat, 7 Mar 2026 18:53:07 -0800 Subject: [PATCH] ci(M01): add ldm.modules.diffusionmodules.openaimodel stub Made-with: Cursor --- scripts/dev/create_stub_repos.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) 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)