fix loading error (#51)

pyreft/config.py

@@ -8,4 +8,24 @@ class ReftConfig(pv.IntervenableConfig):
     def __init__(
         self, **kwargs,
     ):
-        super().__init__(**kwargs)
+        super().__init__(**kwargs)
+
+
+    def to_json_string(self):
+        """
+        Overwrite to bypass trainer initial config checking.
+        
+        If don't overwrite, it may throw json dump error based
+        on your python version.
+        """
+        output = super().to_dict()
+
+        if not isinstance(output["intervention_types"], list):
+            output["intervention_types"] = [output["intervention_types"]]
+        output["intervention_types"] = [
+            str(t) for t in output["intervention_types"]]
+
+        output["representations"] = [
+            str(r) for r in output["representations"]]
+
+        return output

pyreft/interventions.py

@@ -89,7 +89,7 @@ def forward(
 
 
 class ConsreftIntervention(
-    ConstantSourceIntervention,
+    SourcelessIntervention,
     TrainableIntervention, 
     DistributedRepresentationIntervention
 ):