Compute log(alpha_t) of a given continuous-time label t in [0, T].

Add external embeddings to this layer. Use case: >>> 1. Add token to tokenizer and get the token id. >>> tokenizer = Tokeniz

cleanup additional models that were loaded

Return the data prediction model (with thresholding).

embed_dim: output dimension for each position pos: a list of positions to be encoded: size (M,) out: (M, D)

Convert the continuous-time `t_continuous` (in [epsilon, T]) to the model input time. For discrete-time DPMs, we convert `t_continuou

Compute the intermediate time steps for sampling.

A piecewise linear function y = f(x), using xp and yp as keypoints. We implement f(x) in a differentiable way (i.e. applicable for autograd).