Method __init__

models/lumina_2.py:323–329

(self, model)

class FinalLayer(nn.Module):
    def __init__(self, model):
        super().__init__()
        self.final_layer = model.final_layer
        # norm_final isn't used, but by registering it we will keep it in the saved model, preventing ComfyUI from logging a
        # warning that it's missing.
        self.norm_final = model.norm_final
        self.model = [model]

    def __getattr__(self, name):
        return getattr(self.model[0], name)