def forward()

in glide_text2im/clip/encoders.py [0:0]

• 5 lines of code
• 2 McCabe index (conditional complexity)

    def forward(self, x: torch.Tensor, return_probe_features: bool = False) -> torch.Tensor:
        if return_probe_features:
            return x[:, 0]

        x = self.ln(x[:, :1])
        return self.f(x[:, 0])