From ca2de68a739afbe7e3b81de0abef06a30fb6f7c2 Mon Sep 17 00:00:00 2001 From: Michael Yang Date: Wed, 28 Jun 2023 09:24:24 -0700 Subject: [PATCH] add with link --- ollama/cmd/cli.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/ollama/cmd/cli.py b/ollama/cmd/cli.py index f47cdcd24..34d0b91a0 100644 --- a/ollama/cmd/cli.py +++ b/ollama/cmd/cli.py @@ -24,6 +24,7 @@ def main(): generate_parser.set_defaults(fn=generate) add_parser = subparsers.add_parser("add") + add_parser.add_argument("--link", action="store_true") add_parser.add_argument("model") add_parser.set_defaults(fn=add) @@ -57,8 +58,9 @@ def generate(*args, **kwargs): print(choices[0].get("text", ""), end="") -def add(model, models_home): - os.rename(model, Path(models_home) / Path(model).name) +def add(model, models_home, link=False): + fn = os.symlink if link else os.rename + fn(model, Path(models_home) / Path(model).name) def pull(*args, **kwargs):