mindformers.models.glm.GLMChatModel

class mindformers.models.glm.GLMChatModel(config: mindformers.models.glm.glm_config.GLMConfig)

Provide glm chat capability through network. :param config: The config of GLMModel. :type config: GLMConfig

返回

Tensor, the probability distribution of network loss.

sample(log_probs)

Convert the log_probs to probability