matt/ollama37
1
0
Fork 0
mirror of https://github.com/dogkeeper886/ollama37.git synced 2025-12-12 08:47:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

rename model to model_name to avoid collision

Browse Source
This commit is contained in:
Michael Yang
2023-06-28 12:25:08 -07:00
committed by Michael Yang
parent af66c5695a
commit 5cea13ce00
4 changed files with 37 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ollama/cmd/cli.py
View File

@@ -111,7 +111,7 @@ def generate_oneshot(*args, **kwargs):
spinner.start()
spinner_running = True
try:
for output in engine.generate(*args, **kwargs):
for output in engine.generate(model_name=kwargs.pop('model'), *args, **kwargs):
choices = output.get("choices", [])
if len(choices) > 0:
if spinner_running:
@@ -147,7 +147,7 @@ def generate_batch(*args, **kwargs):
def pull(*args, **kwargs):
model.pull(*args, **kwargs)
model.pull(model_name=kwargs.pop('model'), *args, **kwargs)
def run(*args, **kwargs):