Training in progress, step 75000, checkpoint

modeling_novomolgen.py

@@ -263,6 +263,7 @@ class NovoMolGen(GPTLMHeadModel):
         pretrained_model_name_or_path, 
         checkpoint_path: str = "",
         config: Optional[Union[NovoMolGenConfig, str, os.PathLike]] = None,
+        **kwargs,
         ):
         if config is None:
             config = NovoMolGenConfig.from_pretrained(pretrained_model_name_or_path, checkpoint_path=checkpoint_path)