| """Hugging Face configuration for TinyLunarPolicy (32-param linear LunarLander policy).""" | |
| from transformers import PretrainedConfig | |
| class TinyLunarConfig(PretrainedConfig): | |
| model_type = "tiny-lunar-policy" | |
| def __init__(self, obs_dim: int = 8, n_actions: int = 4, bias: bool = False, **kwargs): | |
| self.obs_dim = int(obs_dim) | |
| self.n_actions = int(n_actions) | |
| self.bias = bool(bias) | |
| super().__init__(**kwargs) | |