From 6c574dbecfcbc425082ed5b7f6d74765b18e9b6c Mon Sep 17 00:00:00 2001 From: Deven Mistry <31466137+deven367@users.noreply.github.com> Date: Thu, 21 Dec 2023 09:31:17 -0500 Subject: [PATCH] update path to load weights (#164) --- llms/phi2/phi2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llms/phi2/phi2.py b/llms/phi2/phi2.py index 6e9dd7a2..bcca4209 100644 --- 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(