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.external and sampling stubs (DiscreteEpsDDPMDenoiser, BrownianTreeNoiseSampler, trange)

Browse source 
Made-with: Cursor
This commit is contained in:
Michael Cahill 2026-03-07 18:37:38 -08:00
parent 123a9d2f91
commit feca86bca6
1 changed files with 10 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
scripts/dev/create_stub_repos.py
View file

@ -111,6 +111,13 @@ def main() -> None:
kd = "k-diffusion"
touch(os.path.join(REPOS, kd, "k_diffusion", "__init__.py"))
touch(os.path.join(REPOS, kd, "k_diffusion", "utils.py"), "# stub\n")
touch(
os.path.join(REPOS, kd, "k_diffusion", "external.py"),
"class DiscreteEpsDDPMDenoiser:\n pass\n"
"class DiscreteSchedule:\n pass\n"
"class CompVisVDenoiser:\n pass\n"
"class CompVisDenoiser:\n pass\n",
)
kd_sampling = (
"import torch as _torch\n"
"torch = _torch\n"
@ -118,6 +125,9 @@ def main() -> None:
"def get_sigmas_exponential(*a, **k): pass\n"
"def get_sigmas_polyexponential(*a, **k): pass\n"
"def to_d(*a, **k): pass\n"
"def default_noise_sampler(*a, **k): pass\n"
"def trange(*a, **k): return iter([])\n"
"class BrownianTreeNoiseSampler:\n pass\n"
)
touch(os.path.join(REPOS, kd, "k_diffusion", "sampling.py"), kd_sampling)