wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

nit: docstring

This commit is contained in:
Akshay Kashyap 2024-05-08 10:41:00 -04:00
parent 7461a9131d
commit 3b285e2f26
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/common/policies/diffusion/modeling_diffusion.py
View File

@ -30,7 +30,7 @@ from lerobot.common.policies.utils import (
def _make_noise_scheduler(name: str, **kwargs: dict) -> DDPMScheduler | DDIMScheduler: 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. to the scheduler.
""" """
if name == "DDPM": if name == "DDPM":