ci(M01): add DDPM and register_schedule to ddpm stub

Made-with: Cursor
This commit is contained in:
Michael Cahill 2026-03-07 21:26:21 -08:00
parent bc8de1245d
commit 1e958a4e2a

View file

@ -106,8 +106,10 @@ def main() -> None:
# paths.py asserts ldm/models/diffusion/ddpm.py exists
ddpm_content = (
"# stub for CI - paths.py assertion + LatentDiffusion import\n"
"class LatentDiffusion:\n pass\n"
"# stub for CI - paths.py assertion + LatentDiffusion/DDPM import\n"
"class DDPM:\n"
" def register_schedule(self, *a, **k): pass\n"
"class LatentDiffusion(DDPM):\n pass\n"
"class LatentDepth2ImageDiffusion(LatentDiffusion):\n pass\n"
)
touch(os.path.join(REPOS, sd, "ldm", "models", "diffusion", "ddpm.py"), ddpm_content)