Create configuration_unified_cell.py

configuration_unified_cell.py

@@ -0,0 +1,16 @@
+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 {}