From feca86bca6e97502cb229f222b475ee858885f83 Mon Sep 17 00:00:00 2001 From: Michael Cahill Date: Sat, 7 Mar 2026 18:37:38 -0800 Subject: [PATCH] ci(M01): add k_diffusion.external and sampling stubs (DiscreteEpsDDPMDenoiser, BrownianTreeNoiseSampler, trange) Made-with: Cursor --- scripts/dev/create_stub_repos.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/scripts/dev/create_stub_repos.py b/scripts/dev/create_stub_repos.py index a1e3cfe45..ced2b1197 100644 --- a/scripts/dev/create_stub_repos.py +++ b/scripts/dev/create_stub_repos.py @@ -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)