lets try to change the pipeline

configuration_stacked.py

@@ -1,6 +1,7 @@
 from transformers import PretrainedConfig
 import torch
 
+
 class ImpressoConfig(PretrainedConfig):
     model_type = "stacked_bert"
 
@@ -25,6 +26,7 @@ class ImpressoConfig(PretrainedConfig):
         pretrained_config=None,
         values_override=None,
         label_map=None,
+        filename=None,
         **kwargs,
     ):
         super().__init__(pad_token_id=pad_token_id, **kwargs)
@@ -46,6 +48,7 @@ class ImpressoConfig(PretrainedConfig):
         self.classifier_dropout = classifier_dropout
         self.pretrained_config = pretrained_config
         self.label_map = label_map
+        self.filename = filename
 
         self.values_override = values_override or {}
         self.outputs = {
@@ -72,7 +75,9 @@ class ImpressoConfig(PretrainedConfig):
         """
         return None
 
-    def generate_dummy_inputs(self, tokenizer, batch_size=1, seq_length=8, framework="pt"):
+    def generate_dummy_inputs(
+        self, tokenizer, batch_size=1, seq_length=8, framework="pt"
+    ):
         """
         Generate dummy inputs for testing or export.
         Args:
@@ -88,12 +93,13 @@ class ImpressoConfig(PretrainedConfig):
                 low=0,
                 high=self.vocab_size,
                 size=(batch_size, seq_length),
-                dtype=torch.long
+                dtype=torch.long,
             )
             attention_mask = torch.ones((batch_size, seq_length), dtype=torch.long)
             return {"input_ids": input_ids, "attention_mask": attention_mask}
         else:
             raise ValueError("Framework '{}' not supported.".format(framework))
 
+
 # Register the configuration with the transformers library
 ImpressoConfig.register_for_auto_class()