src/transformers/models/bloom/modeling_bloom.py/BloomGelu/forward
class BloomGelu:    def forward(self, x: torch.Tensor) -> torch.Tensor:
        if self.training:
            return GeLUFunction.apply(x)
        else:
            return bloom_gelu_forward(x)