From c74dd5835d3f560f1f38db0d9d80f5c46e409437 Mon Sep 17 00:00:00 2001 From: Michael Yang Date: Fri, 2 Feb 2024 09:43:36 -0800 Subject: [PATCH] create example --- examples/create/main.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 examples/create/main.py diff --git a/examples/create/main.py b/examples/create/main.py new file mode 100644 index 0000000..0a1161d --- /dev/null +++ b/examples/create/main.py @@ -0,0 +1,20 @@ +import sys + +from ollama import create + + +args = sys.argv[1:] +if len(args) == 2: + # create from local file + path = args[1] +else: + print('usage: python main.py ') + sys.exit(1) + +# TODO: update to real Modelfile values +modelfile = f""" +FROM {path} +""" + +for response in create(model=args[0], modelfile=modelfile, stream=True): + print(response['status'])