Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
small fix
Browse files Browse the repository at this point in the history
rnwang04 committed Aug 9, 2024

Verified

This commit was signed with the committer’s verified signature.
erijo Erik Flodin
1 parent e942222 commit aa1ea6a
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
@@ -30,7 +30,7 @@

if __name__ == '__main__':
parser = argparse.ArgumentParser(description='Predict Tokens using `generate()` API for Llama2 model')
parser.add_argument('--repo-id-or-model-path', type=str, default="TheBloke/Llama-2-7b-Chat-GPTQ",
parser.add_argument('--repo-id-or-model-path', type=str, default="TheBloke/TinyLlama-1.1B-Chat-v1.0-GPTQ",
help='The huggingface repo id'
', or the path to the huggingface checkpoint folder')
parser.add_argument('--prompt', type=str, default="What is AI?",

0 comments on commit aa1ea6a

Please sign in to comment.