Skip to content

GitLab

Commit ea2a453a authored by Michael Yang's avatar Michael Yang
Browse files

fix run arg parser

parent 53c462e4
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
ollama/cmd/cli.py
View file @ ea2a453a
...@@ -33,10 +33,10 @@ def main(): ...@@ -33,10 +33,10 @@ def main():
pull_parser.add_argument("model") pull_parser.add_argument("model")
pull_parser.set_defaults(fn=pull) pull_parser.set_defaults(fn=pull)
pull_parser = subparsers.add_parser("run") run_parser = subparsers.add_parser("run")
pull_parser.add_argument("model") run_parser.add_argument("model")
pull_parser.add_argument("prompt", nargs="?") run_parser.add_argument("prompt", nargs="?")
pull_parser.set_defaults(fn=run) run_parser.set_defaults(fn=run)
args = parser.parse_args() args = parser.parse_args()
args = vars(args) args = vars(args)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment