Remove explicit definition of cfg.inference
This commit is contained in:
committed by
GitHub
parent
bd3b537344
commit
c2508987a6
@@ -182,9 +182,6 @@ def train(
|
||||
if cfg.bf16:
|
||||
cfg.fp16 = True
|
||||
cfg.bf16 = False
|
||||
|
||||
# Store inference mode into cfg when passed via args
|
||||
cfg.inference = True if "inference" in kwargs else cfg.get("inference", False)
|
||||
|
||||
# load the tokenizer first
|
||||
tokenizer_config = cfg.tokenizer_config or cfg.base_model_config
|
||||
|
||||
Reference in New Issue
Block a user