configuration_unified_cell.py

from transformers.configuration_utils import PretrainedConfig

class UnifiedCellClassifierConfig(PretrainedConfig):
    model_type = "unified_cell_classifier"
    
    def __init__(
        self,
        sub_classifier_names=None,
        main_classifier_config=None,
        sub_classifiers_config=None,
        **kwargs
    ):
        super().__init__(**kwargs)
        self.sub_classifier_names = sub_classifier_names or []
        self.main_classifier_config = main_classifier_config or {}
        self.sub_classifiers_config = sub_classifiers_config or {}