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

add function

Browse Source
This commit is contained in:
Bruce MacDonald
2023-06-27 17:36:02 -04:00
parent 7454900733
commit 01d2667f8b
3 changed files with 18 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
ollama/cmd/cli.py
View File

@@ -23,8 +23,8 @@ def main():
generate_parser.set_defaults(fn=generate)
add_parser = subparsers.add_parser("add")
add_parser.add_argument("model_file")
generate_parser.set_defaults(fn=add)
add_parser.add_argument("file")
add_parser.set_defaults(fn=add)
args = parser.parse_args()
args = vars(args)
@@ -48,4 +48,4 @@ def generate(*args, **kwargs):
def add(*args, **kwargs):
model.add(*args, **kwargs)
engine.add(*args, **kwargs)

7
ollama/engine.py
View File

@@ -1,6 +1,7 @@
import os
import json
import sys
import shutil
from contextlib import contextmanager
from llama_cpp import Llama as LLM
from template import template
@@ -61,3 +62,9 @@ def load(model, models_home=".", llms={}):
def unload(model, llms={}):
if model in llms:
llms.pop(model)
def add(file, models_home=".", *args, **kwargs):
if not os.path.exists(file):
raise ValueError("Model file {model} not found")
shutil.move(file, models_home)