fix(config): 修改Checkpoint类中tokenizer和config字段的默认值初始化方式
This commit is contained in:
parent
bad6243b53
commit
12850d403c
|
|
@ -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(
|
||||
|
|
|
|||
Loading…
Reference in New Issue