From 95abb4c4f5b98c5fa5764f9fbd97565abc903159 Mon Sep 17 00:00:00 2001
From: steve <steve@fairfaxmedia.net>
Date: Wed, 13 May 2026 10:26:41 -0400
Subject: [PATCH] fix: increase disk space to 64GB and handle errors during
 tmux session setup

---
 create_payload.json | 2 +-
 vast_start.py       | 5 ++++-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/create_payload.json b/create_payload.json
index c43f86b..81e8446 100644
--- a/create_payload.json
+++ b/create_payload.json
@@ -25,7 +25,7 @@
   "jupyter_dir": null,
   "python_utf8": null,
   "lang_utf8": null,
-  "disk": 32,
+  "disk": 64,
   "last_known_min_bid": 0.6666666666666666,
   "min_duration": 604800
 }
\ No newline at end of file
diff --git a/vast_start.py b/vast_start.py
index e19cb32..a41a9af 100644
--- a/vast_start.py
+++ b/vast_start.py
@@ -175,7 +175,10 @@ def ssh_and_setup(ip: str, port: int, user: str = "root") -> None:
     
     # Disable auto-tmux
     subprocess.run(ssh_base_cmd + ["touch", "~/.no_auto_tmux"], check=True)
-    
+    try:
+        subprocess.run(ssh_base_cmd + ["tmux","new","-d","-s","ollama", "ollama","serve"], check=True)
+    except subprocess.CalledProcessError as e:
+        print(f"Error occurred while setting up tmux session: {e}")
     # Pull the required Ollama model
     pull_cmd = ssh_base_cmd + ["ollama", "pull", OLLAMA_MODEL]
     print(f"Executing: {' '.join(pull_cmd)}")