Skip to content

Commit 3ab5622

Browse files
Update examples/awq/gsm8k_eval.py
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> Signed-off-by: Jagarlamudi <76727507+rtj1@users.noreply.github.com>
1 parent ef02fca commit 3ab5622

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

examples/awq/gsm8k_eval.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,4 +48,8 @@ def evaluate_model(model_path):
4848
print(f"Error: Model path not found: {args.model_path}", file=sys.stderr)
4949
sys.exit(1)
5050

51-
evaluate_model(args.model_path)
51+
if not os.path.isdir(args.model_path):
52+
print(f"Error: Model path not found: {args.model_path}", file=sys.stderr)
53+
sys.exit(1)
54+
55+
evaluate_model(args.model_path)

0 commit comments

Comments
 (0)