lint

mosaicml · Oct 2, 2023 · 256bb01 · 256bb01
1 parent 58fc9d1
commit 256bb01
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/llmfoundry/models/hf/hf_causal_lm.py b/llmfoundry/models/hf/hf_causal_lm.py
@@ -66,7 +66,8 @@ def __init__(self, om_model_config: Union[DictConfig,
                  tokenizer: PreTrainedTokenizerBase):
         # set up training and eval metrics
         use_train_metrics = om_model_config.get('use_train_metrics', True)
-        train_metrics = [LanguageCrossEntropy(), LanguagePerplexity()] if use_train_metrics else []
+        train_metrics = [LanguageCrossEntropy(),
+                         LanguagePerplexity()] if use_train_metrics else []
         eval_metrics = [
             LanguageCrossEntropy(),
             LanguagePerplexity(),

diff --git a/llmfoundry/models/mpt/modeling_mpt.py b/llmfoundry/models/mpt/modeling_mpt.py
@@ -694,8 +694,10 @@ def __init__(
         hf_config = MPTConfig.from_dict(resolved_om_model_config)
         model = MPTForCausalLM(hf_config)
 
-        use_train_metrics = resolved_om_model_config.get('use_train_metrics', True)
-        train_metrics = [LanguageCrossEntropy(), LanguagePerplexity()] if use_train_metrics else []
+        use_train_metrics = resolved_om_model_config.get(
+            'use_train_metrics', True)
+        train_metrics = [LanguageCrossEntropy(),
+                         LanguagePerplexity()] if use_train_metrics else []
         eval_metrics = [
             LanguageCrossEntropy(),
             LanguagePerplexity(),