The input data of a completion request to the LLM. Args: task: The task name. conversation: The conversation history. uui

Output of the score model.

RMSNorm implementation.

Lazily load the json house data.

Tool for making API requests with caching to avoid duplicate calls. Inherits from BaseTool to integrate with eval-anything framework.

Load hdf5_sensors.hdf5 data stored as {dataset_dir}/*/hdf5_sensors.hdf5.

The output data of a completion request to the LLM. Args: task: The task name. uuid: The unique id of the input. engine:

Transformer decoder consisting of *config.num_hidden_layers* layers. Each layer is a [`InternLMDecoderLayer`] Args: config: InternLMC

Lazily load the a list of json houses.

Lazily load a list of json task specs.

The logger utility.

Task sampler has entered some totally invalid state from which next_task calls will definitely fail.

Accustomed Interface for Idefics2 model

Accustomed Interface for LLaVA model

Accustomed Interface for MLLama model

Multi-head attention module.

Multi-head attention module.

Binary cache implementation with support for multiple tensor types

Cache manager for inference results

Trainer for TV2T DPO training.