usse setattr for config assignment

dbanys · Dec 24, 2020 · fe1a0b9 · fe1a0b9
1 parent b8630a3
commit fe1a0b9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/aitextgen/aitextgen.py b/aitextgen/aitextgen.py
@@ -188,13 +188,13 @@ def __init__(
                 )
 
         if gradient_checkpointing:
-            self.model.config["gradient_checkpointing"] = True
+            setattr(self.model.config, "gradient_checkpointing", True)
 
         if schema_tokens:
-            self.model.config["schema_tokens"] = schema_tokens
+            setattr(self.model.config, "schema_tokens", schema_tokens)
 
         if schema_tokens:
-            self.model.config["schema_return"] = schema_return
+            setattr(self.model.config, "schema_return", schema_return)
 
         if self.tokenizer is None:
             # Update tokenizer settings (if not set already)