mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-23 12:40:08 +00:00
refactor(factory, pi0fast): Update processor function names and parameters
- Renamed make_pi0_processor to make_pi0fast_processor for clarity and consistency. - Updated parameter names in the factory's make_processor function to use pretrained_model_name_or_path instead of source, enhancing readability and alignment with naming conventions.
This commit is contained in:
committed by
Steven Palma
parent
7fc7ec75bb
commit
28ef6fcd14
@@ -144,12 +144,12 @@ def make_processor(
|
|||||||
# TODO(azouitine): Handle this case.
|
# TODO(azouitine): Handle this case.
|
||||||
return (
|
return (
|
||||||
RobotProcessor.from_pretrained(
|
RobotProcessor.from_pretrained(
|
||||||
source=pretrained_path,
|
pretrained_model_name_or_path=pretrained_path,
|
||||||
config_filename=kwargs.get("preprocessor_config_filename", "robot_preprocessor.json"),
|
config_filename=kwargs.get("preprocessor_config_filename", "robot_preprocessor.json"),
|
||||||
overrides=kwargs.get("preprocessor_overrides", {}),
|
overrides=kwargs.get("preprocessor_overrides", {}),
|
||||||
),
|
),
|
||||||
RobotProcessor.from_pretrained(
|
RobotProcessor.from_pretrained(
|
||||||
source=pretrained_path,
|
pretrained_model_name_or_path=pretrained_path,
|
||||||
config_filename=kwargs.get("postprocessor_config_filename", "robot_postprocessor.json"),
|
config_filename=kwargs.get("postprocessor_config_filename", "robot_postprocessor.json"),
|
||||||
overrides=kwargs.get("postprocessor_overrides", {}),
|
overrides=kwargs.get("postprocessor_overrides", {}),
|
||||||
),
|
),
|
||||||
|
|||||||
@@ -27,7 +27,7 @@ from lerobot.processor import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
def make_pi0_processor(
|
def make_pi0fast_processor(
|
||||||
config: PI0Config, dataset_stats: dict[str, dict[str, torch.Tensor]] | None = None
|
config: PI0Config, dataset_stats: dict[str, dict[str, torch.Tensor]] | None = None
|
||||||
) -> tuple[RobotProcessor, RobotProcessor]:
|
) -> tuple[RobotProcessor, RobotProcessor]:
|
||||||
input_steps = [
|
input_steps = [
|
||||||
|
|||||||
Reference in New Issue
Block a user