"""Hugging Face configuration for the complete PRIMA model."""
from transformers import LlamaConfig


class PRIMAConfig(LlamaConfig):
    model_type = 'prima'