From e280ac5d719c271042df44c4264e2d49b702d273 Mon Sep 17 00:00:00 2001 From: Michael Cahill Date: Sat, 7 Mar 2026 18:29:56 -0800 Subject: [PATCH] ci(M01): expose sgm.modules.diffusionmodules.model and openaimodel Made-with: Cursor --- scripts/dev/create_stub_repos.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/scripts/dev/create_stub_repos.py b/scripts/dev/create_stub_repos.py index fb6187043..33b9b47fb 100644 --- a/scripts/dev/create_stub_repos.py +++ b/scripts/dev/create_stub_repos.py @@ -68,7 +68,10 @@ def main() -> None: "class CrossAttention:\n def forward(self, *a, **k): pass\n" "\nSDP_IS_AVAILABLE = True\nXFORMERS_IS_AVAILABLE = False\n", ) - touch(os.path.join(REPOS, gm, "sgm", "modules", "diffusionmodules", "__init__.py")) + touch( + os.path.join(REPOS, gm, "sgm", "modules", "diffusionmodules", "__init__.py"), + "from . import model, openaimodel\n", + ) touch( os.path.join(REPOS, gm, "sgm", "modules", "diffusionmodules", "model.py"), "class AttnBlock:\n def forward(self, *a, **k): pass\n",