pali112
/
ollama
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add with link

This commit is contained in:
Michael Yang 2023-06-28 09:24:24 -07:00
parent 52beb0a99e
commit ca2de68a73
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ollama/cmd/cli.py
View File

@ -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):