A fast approximation of the cumulative distribution function of the standard normal.

A fast approximation of the cumulative distribution function of the standard normal.

Compute what the p_mean_variance output would have been, should the model's score function be conditioned by cond_fn. See con

Compute what the p_mean_variance output would have been, should the model's score function be conditioned by cond_fn. See con

Create a logger that writes to a log file and stdout.

Create a logger that writes to a log file and stdout.

删除小数点后多余的0

Get question ids that satisfy given filter conditions. default skips that filter. :param imgIds (int array) : get question ids

r"""A helper function that returns a list of tensor-parallel shard file names by format and tensor parallel size. Args: format (str):

r"""This method infers the checkpoint format and model parallel size according to the files in the given folder. Args: path (str): Th

r"""This function loads tensor parallel checkpoints and handles different formats (e.g., saved by different training frameworks or released by

Compute the KL divergence between two gaussians. Shapes are automatically broadcasted, so batches can be compared to scalars, among other

Compute the KL divergence between two gaussians. Shapes are automatically broadcasted, so batches can be compared to scalars, among other

Diffuse the data for a given number of diffusion steps. In other words, sample from q(x_t | x_0). :param x_start: the initial

Calculates relaxed correctness. The correctness tolerates certain error ratio defined by max_relative_change. See https://arxiv.org/pdf/2203.

Sample a token based on the provided probability distribution using top-p sampling. :param probs: The probability distribution for t

Convert model output scores into ranks.