Fix AttributeError when finish training

urchade · Jul 31, 2024 · df16c54 · df16c54
1 parent 3359b60
commit df16c54
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/gliner/model.py b/gliner/model.py
@@ -81,6 +81,10 @@ def __init__(self, config: GLiNERConfig,
         else:
             self.onnx_model = False
 
+        # to suppress an AttributeError when training
+        self._keys_to_ignore_on_save = None
+
+
     def forward(self, *args, **kwargs):
         """Wrapper function for the model's forward pass."""
         output = self.model(*args, **kwargs)