| """``transformers`` configuration of encoder-tvsd; the model itself is in ``modeling.py``. |
| |
| ``config.json`` describes every variant: the shared fields, ``default_variant``, and ``variants`` |
| (each variant's ``members``, ``areas`` and ``scores``). There is deliberately no ``variant`` |
| attribute: ``AutoModel.from_pretrained(..., variant="tiny")`` must reach the model class, which picks |
| the variant and its weight file. |
| """ |
|
|
| from transformers import PretrainedConfig |
|
|
|
|
| class EncoderTVSDConfig(PretrainedConfig): |
| model_type = "encoder_tvsd" |
|
|
| def __init__( |
| self, |
| default_variant: str = "", |
| variants: dict | None = None, |
| monkeys: list | None = None, |
| electrode_index: dict | None = None, |
| **kwargs, |
| ): |
| super().__init__(**kwargs) |
| self.default_variant = default_variant |
| self.variants = variants or {} |
| self.monkeys = monkeys or [] |
| self.electrode_index = electrode_index or {} |
|
|
