config

Base configuration class for Lightning IR models.

This module defines the configuration class LightningIRConfig which is used to instantiate a Lightning IR model. The configuration class acts as a mixin for the transformers.PretrainedConfig class from the Hugging Face Transformers library.

Classes

LightningIRConfig(*args[, query_length, ...])

The configuration class to instantiate a Lightning IR model.