configuration_ohca.py

# configuration_ohca.py
from transformers import PretrainedConfig

class OhcaConfig(PretrainedConfig):
    model_type = "cvd_ohca_predictor"

    def __init__(self, **kwargs):
        super().__init__(**kwargs)