File size: 585 Bytes
1bfc3f0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 | """Arbitrary-horizon frontend bound to one ActionCodec2 action space."""
from transformers import AutoConfig, AutoProcessor
from ...backends.actioncodec2.hf import ActionCodec2FrontendHFConfig
from .generation import ActionCodec2LogitsProcessor
from .processor import ActionCodec2
from .tokenizer import ActionCodec2Tokenizer
AutoConfig.register(ActionCodec2.model_type, ActionCodec2FrontendHFConfig, exist_ok=True)
AutoProcessor.register(ActionCodec2FrontendHFConfig, ActionCodec2, exist_ok=True)
__all__ = ["ActionCodec2", "ActionCodec2Tokenizer", "ActionCodec2LogitsProcessor"]
|