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

ci(M01): add k_diffusion.sampling stubs (get_sigmas_*, torch)

Browse source 
Made-with: Cursor
This commit is contained in:
Michael Cahill 2026-03-07 18:32:23 -08:00
parent e280ac5d71
commit f1a2fd9866
1 changed files with 10 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

12
scripts/dev/create_stub_repos.py
View file

@ -107,10 +107,18 @@ def main() -> None:
"# stub\n",
)
# k-diffusion: k_diffusion.sampling
# k-diffusion: k_diffusion.sampling (sd_schedulers, sd_samplers_common)
kd = "k-diffusion"
touch(os.path.join(REPOS, kd, "k_diffusion", "__init__.py"))
touch(os.path.join(REPOS, kd, "k_diffusion", "sampling.py"), "# stub\n")
kd_sampling = (
"import torch as _torch\n"
"torch = _torch\n"
"def get_sigmas_karras(*a, **k): pass\n"
"def get_sigmas_exponential(*a, **k): pass\n"
"def get_sigmas_polyexponential(*a, **k): pass\n"
"def to_d(*a, **k): pass\n"
)
touch(os.path.join(REPOS, kd, "k_diffusion", "sampling.py"), kd_sampling)
# BLIP: models/blip.py
touch(os.path.join(REPOS, "BLIP", "models", "blip.py"), "# stub\n")