From 62cd96ea687ee2b6319dff89f66cdacdcc54a742 Mon Sep 17 00:00:00 2001
From: Lucas Armand <lucas@vast.ai>
Date: Tue, 24 Mar 2026 13:09:04 -0700
Subject: [PATCH] Allow pre-release

---
 start_server.sh | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/start_server.sh b/start_server.sh
index c325722..248bc32 100755
--- a/start_server.sh
+++ b/start_server.sh
@@ -53,7 +53,7 @@ function install_vastai_sdk() {
             echo "WARNING: Both SDK_BRANCH and SDK_VERSION are set; using SDK_BRANCH=${SDK_BRANCH}"
         fi
         echo "Installing vastai-sdk from https://github.com/vast-ai/vast-sdk/ @ ${SDK_BRANCH}"
-        if ! uv pip install "vastai-sdk @ git+https://github.com/vast-ai/vast-sdk.git@${SDK_BRANCH}"; then
+        if ! uv pip install --prerelease=allow "vastai-sdk @ git+https://github.com/vast-ai/vast-sdk.git@${SDK_BRANCH}"; then
             report_error_and_exit "Failed to install vastai-sdk from vast-ai/vast-sdk@${SDK_BRANCH}"
         fi
         return 0
@@ -61,14 +61,14 @@ function install_vastai_sdk() {
 
     if [ -n "${SDK_VERSION:-}" ]; then
         echo "Installing vastai-sdk version ${SDK_VERSION}"
-        if ! uv pip install "vastai-sdk==${SDK_VERSION}"; then
+        if ! uv pip install --prerelease=allow "vastai-sdk==${SDK_VERSION}"; then
             report_error_and_exit "Failed to install vastai-sdk==${SDK_VERSION}"
         fi
         return 0
     fi
 
     echo "Installing default vastai-sdk"
-    if ! uv pip install vastai-sdk; then
+    if ! uv pip install --prerelease=allow vastai-sdk; then
         report_error_and_exit "Failed to install vastai-sdk"
     fi
 }