admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-24 02:06:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

guard molmoact2 processor transformers import

Browse Source
This commit is contained in:
hq-fang
2026-05-19 21:59:00 +00:00
parent e13c6a6110
commit c3c9c2b089
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/molmoact2/processor_molmoact2.py
+7 -3
View File
@@ -23,7 +23,7 @@ from contextlib import suppress
from copy import deepcopy from copy import deepcopy
from dataclasses import dataclass, field from dataclasses import dataclass, field
from pathlib import Path from pathlib import Path
from typing import Any from typing import TYPE_CHECKING, Any
import numpy as np import numpy as np
import torch import torch
@@ -52,10 +52,15 @@ from lerobot.utils.constants import (
POLICY_POSTPROCESSOR_DEFAULT_NAME, POLICY_POSTPROCESSOR_DEFAULT_NAME,
POLICY_PREPROCESSOR_DEFAULT_NAME, POLICY_PREPROCESSOR_DEFAULT_NAME,
) )
from lerobot.utils.import_utils import require_package from lerobot.utils.import_utils import _transformers_available, require_package
from .configuration_molmoact2 import MolmoAct2Config, infer_molmoact2_max_sequence_length from .configuration_molmoact2 import MolmoAct2Config, infer_molmoact2_max_sequence_length
if TYPE_CHECKING or _transformers_available:
from transformers import AutoProcessor
else:
AutoProcessor = None
ACTION_OUTPUT_TOKEN = "<action_output>" # nosec B105 ACTION_OUTPUT_TOKEN = "<action_output>" # nosec B105
ACTION_START_TOKEN = "<action_start>" # nosec B105 ACTION_START_TOKEN = "<action_start>" # nosec B105
ACTION_END_TOKEN = "<action_end>" # nosec B105 ACTION_END_TOKEN = "<action_end>" # nosec B105
@@ -509,7 +514,6 @@ class MolmoAct2PackInputsProcessorStep(ProcessorStep):
def __post_init__(self) -> None: def __post_init__(self) -> None:
require_package("transformers", extra="molmoact2") require_package("transformers", extra="molmoact2")
from transformers import AutoProcessor
checkpoint_location = _resolve_checkpoint_location( checkpoint_location = _resolve_checkpoint_location(
self.checkpoint_path, self.checkpoint_path,