update path to load weights (#164)

2025-09-01 04:14:38 +08:00 · 2023-12-21 09:31:17 -05:00
parent 4addd02988
commit 6c574dbecf
1 changed files with 1 additions and 1 deletions
--- a/llms/phi2/phi2.py
+++ b/llms/phi2/phi2.py
@@ -169,7 +169,7 @@ if __name__ == "__main__":
    parser.add_argument(
        "--model-path",
        type=str,
-        default="phi-2",
+        default=".",
        help="The path to the model weights",
    )
    parser.add_argument(