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(