fix: increase disk space to 64GB and handle errors during tmux session setup

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
 }

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)}")