mindformers.models.glm.GLMChatModel

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

Provide glm chat capability through network. Args:

config (GLMConfig): The config of GLMModel.

Returns:

Tensor, the probability distribution of network loss.

sample(log_probs)

Convert the log_probs to probability