model path updated
Browse files- scripts/eval.py +1 -1
scripts/eval.py
CHANGED
|
@@ -22,7 +22,7 @@ def parse_args():
|
|
| 22 |
parser = argparse.ArgumentParser(description="Evaluate myolmoe model")
|
| 23 |
|
| 24 |
# Model arguments
|
| 25 |
-
parser.add_argument("--model_path", type=str, default="/
|
| 26 |
help="Path to the pretrained model")
|
| 27 |
parser.add_argument("--model_type", type=str, default="hf-auto",
|
| 28 |
help="Model type for lm-eval")
|
|
|
|
| 22 |
parser = argparse.ArgumentParser(description="Evaluate myolmoe model")
|
| 23 |
|
| 24 |
# Model arguments
|
| 25 |
+
parser.add_argument("--model_path", type=str, default="/root/.cache/huggingface/hub/models--allenai--OLMoE-7B/snapshots/6d84c48581ece794365f2b8e9cfb043c68ade9c5",
|
| 26 |
help="Path to the pretrained model")
|
| 27 |
parser.add_argument("--model_type", type=str, default="hf-auto",
|
| 28 |
help="Model type for lm-eval")
|