(cls, model_class: Type, metadata: AttentionProcessorMetadata)
| 59 | |
| 60 | @classmethod |
| 61 | def register(cls, model_class: Type, metadata: AttentionProcessorMetadata): |
| 62 | cls._register() |
| 63 | cls._registry[model_class] = metadata |
| 64 | |
| 65 | @classmethod |
| 66 | def get(cls, model_class: Type) -> AttentionProcessorMetadata: |