mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2026-03-22 14:20:39 -07:00
Code Issues Projects Releases Packages Wiki Activity

M01: add sgm denoiser_scaling, discretizer, DiffusionEngine, GeneralConditioner stubs; M01_run3 and M01_summary

Browse source 
Made-with: Cursor
This commit is contained in:
Michael Cahill 2026-03-07 16:25:26 -08:00
parent 1d3c4dcbd5
commit bc3bd812dd
3 changed files with 188 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

31
scripts/dev/create_stub_repos.py
View file

@ -65,7 +65,8 @@ def main() -> None:
# sgm.modules.attention, diffusionmodules.model (sd_hijack_optimizations)
touch(
os.path.join(REPOS, gm, "sgm", "modules", "attention", "__init__.py"),
"class CrossAttention:\n def forward(self, *a, **k): pass\n",
"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(
@ -74,7 +75,33 @@ def main() -> None:
)
# 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"))
touch(
os.path.join(REPOS, gm, "sgm", "models", "diffusion", "__init__.py"),
"class DiffusionEngine:\n pass\n",
)
# sgm.modules.diffusionmodules.denoiser_scaling, discretizer (sd_models_xl)
touch(
os.path.join(REPOS, gm, "sgm", "modules", "diffusionmodules", "denoiser_scaling.py"),
"class VScaling:\n pass\n",
)
touch(
os.path.join(REPOS, gm, "sgm", "modules", "diffusionmodules", "discretizer.py"),
"class LegacyDDPMDiscretization:\n alphas_cumprod = [1.0]\n",
)
# sgm.modules.GeneralConditioner (sd_models_xl)
touch(os.path.join(REPOS, gm, "sgm", "modules", "__init__.py"))
touch(
os.path.join(REPOS, gm, "sgm", "modules", "conditioner.py"),
"class GeneralConditioner:\n pass\n",
)
touch(
os.path.join(REPOS, gm, "sgm", "modules", "__init__.py"),
"from .conditioner import GeneralConditioner\n",
)
touch(
os.path.join(REPOS, gm, "sgm", "modules", "diffusionmodules", "openaimodel.py"),
"# stub\n",
)
# k-diffusion: k_diffusion.sampling
kd = "k-diffusion"