Method register

src/diffusers/hooks/_helpers.py:88–91

(cls, model_class: Type, metadata: TransformerBlockMetadata)

    @classmethod
    def register(cls, model_class: Type, metadata: TransformerBlockMetadata):
        cls._register()
        metadata._cls = model_class
        cls._registry[model_class] = metadata

    @classmethod
    def get(cls, model_class: Type) -> TransformerBlockMetadata: