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

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(