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
to the scheduler.