Add speaker and start/end signal on each round.

compute the accuracy (soft score) of human answers

Converts a BLOOM Causal LM to a Prefix LM. Supported HuggingFace model classes: - `BloomForCausalLM` See `convert_hf_causal_lm_to_pr

Converts a GPT-style Causal LM to a Prefix LM. Supported HuggingFace model classes: - `GPT2LMHeadModel` - `GPTNeoForCausalLM`

Converts an OPT Causal LM to a Prefix LM. Supported HuggingFace model classes: - `OPTForCausalLM` See `convert_hf_causal_lm_to_prefi

Adds sentinel tokens and padding token (if missing). Expands the tokenizer vocabulary to include sentinel tokens used in mixture-of-denoiser

Create a 1D, 2D, or 3D average pooling module.

Build the sampling part of a task queue @param task_queue: the target task queue @param net: the network @param is_decoder: currently

Build a single task queue for the encoder or decoder @param net: the VAE decoder or encoder network @param is_decoder: currently building

Crop the valid region from the tile @param x: input tile @param input_bbox: original input bounding box @param target_bbox: output bo

create a function from a single tile without summary

Get inverse affine matrix.