Method __init__

models/cosmos.py:403–406

(self, pipeline)

class FinalLayer(nn.Module):
    def __init__(self, pipeline):
        super().__init__()
        self.pipeline = pipeline
        self.final_layer = self.pipeline.transformer.final_layer

    def __getattr__(self, name):
        return getattr(self.pipeline.transformer, name)