Method __init__

models/cosmos_predict2.py:578–580

(self, llm_adapter)

class LLMAdapterLayer(nn.Module):
    def __init__(self, llm_adapter):
        super().__init__()
        self.llm_adapter = llm_adapter

    @torch.autocast('cuda', dtype=AUTOCAST_DTYPE)
    def forward(self, inputs):