From aca1d06a25d6dbcc54f8b2f9aad12d737aa0134c Mon Sep 17 00:00:00 2001 From: S Sitaraman Date: Tue, 27 Aug 2024 03:36:14 +0530 Subject: [PATCH] Inconsistent indentation fix in get_train_stats function --- utils/validate_data.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/validate_data.py b/utils/validate_data.py index 2632eaf..db7c55c 100644 --- a/utils/validate_data.py +++ b/utils/validate_data.py @@ -35,7 +35,7 @@ "open-mixtral-8x7b": 2966, "open-mixtral-8x22b": 1007, "mistral-large-latest": 567, - 'open-mistral-nemo': 3337, + "open-mistral-nemo": 3337, } MIN_NUM_JSONL_LINES = 10 @@ -110,7 +110,7 @@ def get_train_stats( model_id = "mistral-large-latest" elif params_config["dim"] == 5120: model_id = "open-mistral-nemo" - else: + else: raise ValueError("Provided model folder seems incorrect.") else: model_id = train_args.model_id_or_path