Given source instruction and response pieces, return the text containing the complete conversation, and the list of values that the m

Apply rotary embeddings to input tensors using the given frequency tensor. This function applies rotary embeddings to the gi

Apply rotary embeddings to input tensors using the given frequency tensor. This function applies rotary embeddings to the gi

Create a beta schedule that discretizes the given alpha_t_bar function, which defines the cumulative product of (1-beta) over time from t = [

Create a beta schedule that discretizes the given alpha_t_bar function, which defines the cumulative product of (1-beta) over time from t = [

End DDP training.

End DDP training.

Clips the gradient norm of all parameters. The norm is computed over all parameters' gradients as viewed as a single vector, and the grad

Compute logits for a given list of text examples or token lists, optionally incorporating images. :param examples: A batched list of

Compute the mean for the previous step, given a function cond_fn that computes the gradient of a conditional log probability with res

Compute the mean for the previous step, given a function cond_fn that computes the gradient of a conditional log probability with res

Sample x_{t-1} from the model using DDIM. Same usage as p_sample().

Sample x_{t-1} from the model using DDIM. Same usage as p_sample().

Use DDIM to sample from the model and yield intermediate samples from each timestep of DDIM. Same usage as p_sample_loop_prog

Use DDIM to sample from the model and yield intermediate samples from each timestep of DDIM. Same usage as p_sample_loop_prog