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

kwargs fix

This commit is contained in:
Akshay Kashyap 2024-05-08 10:36:27 -04:00
parent b54d05276c
commit 7461a9131d
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

@ -28,7 +28,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 given name. All kwargs are passed
to the scheduler. to the scheduler.