kanshan/ollama-python
1
0
Fork 0
mirror of https://github.com/ollama/ollama-python.git synced 2026-05-02 12:18:18 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

accept license, template, system as files

Browse Source
This commit is contained in:
Michael Yang 2024-05-08 12:53:35 -07:00
parent e403d748a3
commit 59c0542fa1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ollama/_client.py
View File

@ -286,7 +286,7 @@ class Client(BaseClient):
out = io.StringIO() out = io.StringIO()
for line in io.StringIO(modelfile): for line in io.StringIO(modelfile):
command, _, args = line.partition(' ') command, _, args = line.partition(' ')
if command.upper() not in ['FROM', 'ADAPTER']: if command.upper() not in ['FROM', 'ADAPTER', 'LICENSE', 'TEMPLATE', 'SYSTEM']:
print(line, end='', file=out) print(line, end='', file=out)
continue continue
@ -568,7 +568,7 @@ class AsyncClient(BaseClient):
out = io.StringIO() out = io.StringIO()
for line in io.StringIO(modelfile): for line in io.StringIO(modelfile):
command, _, args = line.partition(' ') command, _, args = line.partition(' ')
if command.upper() not in ['FROM', 'ADAPTER']: if command.upper() not in ['FROM', 'ADAPTER', 'LICENSE', 'TEMPLATE', 'SYSTEM']:
print(line, end='', file=out) print(line, end='', file=out)
continue continue