From 28d006b1abc9584237f0ccbca58d712cfd61ecaa Mon Sep 17 00:00:00 2001 From: Michael Cahill Date: Sat, 7 Mar 2026 18:27:23 -0800 Subject: [PATCH] ci(M01): fix sgm.modules.attention attribute - explicit submodule imports Made-with: Cursor --- scripts/dev/create_stub_repos.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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"),