Upload folder using huggingface_hub

config.json

@@ -8,9 +8,13 @@
   "sampling_name": "high-disagreement@100+nonstratified",
   "number_item": "100",
   "fitted_model_type": "RandomForestRegressor_100",
+  "use_full_prompt": true,
+  "data_path": "arubique/flattened-MMLU",
   "eval_config": {
     "pca": 256,
     "pad_to_size": 31,
-    "use_lmeval_batching": true
+    "use_lmeval_batching": true,
+    "use_full_prompt": true,
+    "data_path": "arubique/flattened-MMLU"
   }
 }

configuration_disco.py

@@ -15,6 +15,8 @@ class DiscoConfig(PretrainedConfig):
         sampling_name: str = "",
         number_item: str = "",
         fitted_model_type: str = "",
+        use_full_prompt: bool = True,
+        data_path: str = "",
         **kwargs,
     ):
         super().__init__(**kwargs)
@@ -22,3 +24,5 @@ class DiscoConfig(PretrainedConfig):
         self.sampling_name = sampling_name
         self.number_item = number_item
         self.fitted_model_type = fitted_model_type
+        self.use_full_prompt = use_full_prompt
+        self.data_path = data_path