File size: 449 Bytes
6b4df6d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
from transformers import LlavaConfig
from typing import Dict, List, Optional
class Bitvla_Config(LlavaConfig):
model_type: str = "bitvla"
def __init__(
self,
norm_stats: Optional[Dict[str, Dict[str, Dict[str, Dict[str, List[float]]]]]] = None,
n_action_bins: int = 256,
**kwargs: str,
) -> None:
self.norm_stats, self.n_action_bins = norm_stats, n_action_bins
super().__init__(**kwargs)
|