create models home if it does not exist

e4bdacb1 · Bruce MacDonald · 28b47e60 · e4bdacb1
Commit e4bdacb1 authored Jun 29, 2023 by Bruce MacDonald
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

ollama/cmd/cli.py ollama/cmd/cli.py +5 -0

No files found.
--- a/ollama/cmd/cli.py
+++ b/ollama/cmd/cli.py
@@ -11,6 +11,11 @@ def main():
    parser = ArgumentParser()
    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()

    server.set_parser(subparsers.add_parser("serve"))