Update configuration_stacked.py

configuration_stacked.py

@@ -25,7 +25,6 @@ class ImpressoConfig(PretrainedConfig):
         pretrained_config=None,
         values_override=None,
         label_map=None,
-        outputs=False,
         **kwargs,
     ):
         super().__init__(pad_token_id=pad_token_id, **kwargs)
@@ -49,7 +48,9 @@ class ImpressoConfig(PretrainedConfig):
         self.label_map = label_map
 
         self.values_override = values_override or {}
-        self.outputs = outputs
+        self.outputs = {
+            "logits": {"shape": [None, None, self.hidden_size], "dtype": "float32"}
+        }
 
     @classmethod
     def is_torch_support_available(cls):