diff --git a/khaosz/config/param_config.py b/khaosz/config/param_config.py index de8fd13..1a3b9f6 100644 --- a/khaosz/config/param_config.py +++ b/khaosz/config/param_config.py @@ -100,11 +100,11 @@ class Checkpoint(BaseModelIO): metadata={"help": "Transformer model."} ) tokenizer: BpeTokenizer = field( - default=None, + default_factory=BpeTokenizer, metadata={"help": "Tokenizer for the model."} ) config: TransformerConfig = field( - default=None, + default_factory=TransformerConfig, metadata={"help": "Transformer model configuration."} ) optimizer_state: Dict[str, Any] = field(