mindformers.models.bloom.BloomLMHeadModel¶

class mindformers.models.bloom.BloomLMHeadModel(config=None)[源代码]¶

Provide bloom training loss or logits through network.

Args:: config (BloomConfig): The config of BloomModel.
Returns:: Tensor, the loss or logits of the network.

get_top_token_id(logits, current_index=None)[源代码]¶