ci(M01): fix sgm.modules.attention attribute - explicit submodule imports

Made-with: Cursor
This commit is contained in:
Michael Cahill 2026-03-07 18:27:23 -08:00
parent bc3bd812dd
commit 28d006b1ab

View file

@ -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"),