From 1d3c4dcbd5cd2f3ee339b1eb4d63ba7c8ba6bbbd Mon Sep 17 00:00:00 2001
From: Michael Cahill <michael.cahill@ymail.com>
Date: Sat, 7 Mar 2026 16:22:39 -0800
Subject: [PATCH] M01: add sgm.models.diffusion stub

Made-with: Cursor
---
 scripts/dev/create_stub_repos.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/scripts/dev/create_stub_repos.py b/scripts/dev/create_stub_repos.py
index 7f8390a91..0012157c0 100644
--- a/scripts/dev/create_stub_repos.py
+++ b/scripts/dev/create_stub_repos.py
@@ -72,6 +72,9 @@ def main() -> None:
         os.path.join(REPOS, gm, "sgm", "modules", "diffusionmodules", "model.py"),
         "class AttnBlock:\n    def forward(self, *a, **k): pass\n",
     )
+    # sgm.models.diffusion (sd_models_xl)
+    touch(os.path.join(REPOS, gm, "sgm", "models", "__init__.py"))
+    touch(os.path.join(REPOS, gm, "sgm", "models", "diffusion", "__init__.py"))
 
     # k-diffusion: k_diffusion.sampling
     kd = "k-diffusion"