debug

configuration_stacked.py

@@ -4,16 +4,16 @@ import torch
 class ImpressoConfig(PretrainedConfig):
     model_type = "floret"
 
-    # def __init__(self, filename="LID-40-3-2000000-1-4.bin", **kwargs):
-    #     super().__init__(**kwargs)
-    #     self.filename = filename
+    def __init__(self, filename="LID-40-3-2000000-1-4.bin", **kwargs):
+        super().__init__(**kwargs)
+        self.filename = filename
 
-    # @classmethod
-    # def from_pretrained(cls, pretrained_model_name_or_path, **kwargs):
-    #     # Bypass JSON loading and create config directly
-    #     print(f"Loading ImpressoConfig from {pretrained_model_name_or_path}")
-    #     config = cls(filename="LID-40-3-2000000-1-4.bin", **kwargs)
-    #     return config
+    @classmethod
+    def from_pretrained(cls, pretrained_model_name_or_path, **kwargs):
+        # Bypass JSON loading and create config directly
+        print(f"Loading ImpressoConfig from {pretrained_model_name_or_path}")
+        config = cls(filename="LID-40-3-2000000-1-4.bin", **kwargs)
+        return config