Upload evaluation/benchmarks/code_generation/eval.py with huggingface_hub
Browse files
evaluation/benchmarks/code_generation/eval.py
CHANGED
|
@@ -11,9 +11,8 @@ def main():
|
|
| 11 |
parser.add_argument("model_path", type=str, help="Path to model checkpoint")
|
| 12 |
args = parser.parse_args()
|
| 13 |
|
| 14 |
-
|
| 15 |
-
|
| 16 |
-
print(f"Error: Path not found at '{args.model_path}'", file=sys.stderr)
|
| 17 |
sys.exit(1)
|
| 18 |
|
| 19 |
checkpoint_name = os.path.basename(os.path.normpath(args.model_path))
|
|
|
|
| 11 |
parser.add_argument("model_path", type=str, help="Path to model checkpoint")
|
| 12 |
args = parser.parse_args()
|
| 13 |
|
| 14 |
+
if not os.path.isfile(args.model_path):
|
| 15 |
+
print(f"Error: Directory not found at '{args.model_path}'", file=sys.stderr)
|
|
|
|
| 16 |
sys.exit(1)
|
| 17 |
|
| 18 |
checkpoint_name = os.path.basename(os.path.normpath(args.model_path))
|