baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fix run arg parser

Browse source
This commit is contained in:
Michael Yang 2023-06-29 16:32:00 -07:00
parent 53c462e40d
commit ea2a453a81
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
ollama/cmd/cli.py
View file

@ -33,10 +33,10 @@ def main():
pull_parser.add_argument("model")
pull_parser.set_defaults(fn=pull)
pull_parser = subparsers.add_parser("run")
pull_parser.add_argument("model")
pull_parser.add_argument("prompt", nargs="?")
pull_parser.set_defaults(fn=run)
run_parser = subparsers.add_parser("run")
run_parser.add_argument("model")
run_parser.add_argument("prompt", nargs="?")
run_parser.set_defaults(fn=run)
args = parser.parse_args()
args = vars(args)