PRIMA / code /model /configuration_prima.py
kanguyen-vn's picture
Include clean PRIMA code and evaluator; document metrics, initialization, and MIT contributions
91c1e9e verified
Raw History Blame Contribute Delete
160 Bytes
"""Hugging Face configuration for the complete PRIMA model."""
from transformers import LlamaConfig
class PRIMAConfig(LlamaConfig):
model_type = 'prima'