update path to load weights (#164)

This commit is contained in:
Deven Mistry 2023-12-21 09:31:17 -05:00 committed by GitHub
parent 4addd02988
commit 6c574dbecf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -169,7 +169,7 @@ if __name__ == "__main__":
parser.add_argument( parser.add_argument(
"--model-path", "--model-path",
type=str, type=str,
default="phi-2", default=".",
help="The path to the model weights", help="The path to the model weights",
) )
parser.add_argument( parser.add_argument(