Fix tokenizer missing form config.to_dict() (#228)

Also orders alphabetically
CarperAI · Jan 27, 2023 · aeab7cc · aeab7cc
1 parent ce5bf57
commit aeab7cc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/trlx/data/configs.py b/trlx/data/configs.py
@@ -217,9 +217,9 @@ class TRLConfig:
 
     method: MethodConfig
     model: ModelConfig
-    tokenizer: TokenizerConfig
     optimizer: OptimizerConfig
     scheduler: SchedulerConfig
+    tokenizer: TokenizerConfig
     train: TrainConfig
 
     @classmethod
@@ -243,6 +243,7 @@ def to_dict(self):
             "model": self.model.__dict__,
             "optimizer": self.optimizer.__dict__,
             "scheduler": self.scheduler.__dict__,
+            "tokenizer": self.tokenizer.__dict__,
             "train": self.train.__dict__,
         }