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(
"--model-path",
type=str,
default="phi-2",
default=".",
help="The path to the model weights",
)
parser.add_argument(