(cls, model_class: Type)
| 92 | |
| 93 | @classmethod |
| 94 | def get(cls, model_class: Type) -> TransformerBlockMetadata: |
| 95 | cls._register() |
| 96 | if model_class not in cls._registry: |
| 97 | raise ValueError(f"Model class {model_class} not registered.") |
| 98 | return cls._registry[model_class] |
| 99 | |
| 100 | @classmethod |
| 101 | def _register(cls): |