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:
|
if cfg.bf16:
|
||||||
cfg.fp16 = True
|
cfg.fp16 = True
|
||||||
cfg.bf16 = False
|
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
|
# load the tokenizer first
|
||||||
tokenizer_config = cfg.tokenizer_config or cfg.base_model_config
|
tokenizer_config = cfg.tokenizer_config or cfg.base_model_config
|
||||||
|
|||||||
Reference in New Issue
Block a user