Update model loading exception message

Signed-off-by: Kelly A <[email protected]>
kellyaa · May 14, 2024 · 90b98f3 · 90b98f3
1 parent cd260a3
commit 90b98f3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/build/launch_training.py b/build/launch_training.py
@@ -109,7 +109,9 @@ def main():
             sys.exit(USER_ERROR_EXIT_CODE)
         except HFValidationError as e:
             logging.error(traceback.format_exc())
-            write_termination_log(f"Specified base model not found. Exception: {e}")
+            write_termination_log(
+                f"There may be a problem with loading the model. Exception: {e}"
+            )
             sys.exit(USER_ERROR_EXIT_CODE)
         except (TypeError, ValueError, EnvironmentError) as e:
             logging.error(traceback.format_exc())