diff --git a/lerobot/common/policies/diffusion/modeling_diffusion.py b/lerobot/common/policies/diffusion/modeling_diffusion.py index 8005b33d..3cf0516e 100644 --- a/lerobot/common/policies/diffusion/modeling_diffusion.py +++ b/lerobot/common/policies/diffusion/modeling_diffusion.py @@ -30,7 +30,7 @@ from lerobot.common.policies.utils import ( def _make_noise_scheduler(name: str, **kwargs: dict) -> DDPMScheduler | DDIMScheduler: """ - Factory for noise scheduler instances of the given name. All kwargs are passed + Factory for noise scheduler instances of the requested type. All kwargs are passed to the scheduler. """ if name == "DDPM":