tcsenpai/ollama
1
0
Fork 0
mirror of https://github.com/tcsenpai/ollama.git synced 2025-06-07 03:35:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

create models home if it does not exist

Browse Source
This commit is contained in:
Bruce MacDonald 2023-06-29 15:20:08 -04:00
parent 28b47e6068
commit e4bdacb152
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ollama/cmd/cli.py
View File

@ -11,6 +11,11 @@ def main():
parser = ArgumentParser() parser = ArgumentParser()
parser.add_argument("--models-home", default=Path.home() / ".ollama" / "models") parser.add_argument("--models-home", default=Path.home() / ".ollama" / "models")
# create models home if it doesn't exist
models_home = parser.parse_known_args()[0].models_home
if not models_home.exists():
os.makedirs(models_home)
subparsers = parser.add_subparsers() subparsers = parser.add_subparsers()
server.set_parser(subparsers.add_parser("serve")) server.set_parser(subparsers.add_parser("serve"))