Angainor Development commited on
Commit
c250898
·
unverified ·
1 Parent(s): bd3b537

Remove explicit definition of cfg.inference

Browse files
Files changed (1) hide show
  1. scripts/finetune.py +0 -3
scripts/finetune.py CHANGED
@@ -182,9 +182,6 @@ def train(
182
  if cfg.bf16:
183
  cfg.fp16 = True
184
  cfg.bf16 = False
185
-
186
- # Store inference mode into cfg when passed via args
187
- cfg.inference = True if "inference" in kwargs else cfg.get("inference", False)
188
 
189
  # load the tokenizer first
190
  tokenizer_config = cfg.tokenizer_config or cfg.base_model_config
 
182
  if cfg.bf16:
183
  cfg.fp16 = True
184
  cfg.bf16 = False
 
 
 
185
 
186
  # load the tokenizer first
187
  tokenizer_config = cfg.tokenizer_config or cfg.base_model_config