File size: 265 Bytes
9b4302d | 1 2 3 4 5 6 7 | """Model definitions and loaders for GPT-2 LLM2Vec conversion."""
from gpt2_llm2vec.models.bidirectional_gpt2 import GPT2BiForMNTP, GPT2BiModel
from gpt2_llm2vec.models.registry import get_model_class
__all__ = ["GPT2BiModel", "GPT2BiForMNTP", "get_model_class"]
|