From 037d774f2cd549a5071345e89f70192ca6f4d051 Mon Sep 17 00:00:00 2001
From: erew123 <35898566+erew123@users.noreply.github.com>
Date: Wed, 12 Jun 2024 17:13:18 +0100
Subject: [PATCH] Update re Google Colab/DeepSpeed compile

DeepSpeed needs to compile the first time it loads in and a longer wait time is required for the 1st model load with DeepSpeed
---
 script.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/script.py b/script.py
index fbd37b9..621ca3b 100644
--- a/script.py
+++ b/script.py
@@ -1408,7 +1408,7 @@ if gradio_enabled == True:
         # Restart the subprocess
         restart_subprocess()
         # Wait for the engine to be ready with error handling and retries
-        max_retries = 80
+        max_retries = 140
         retry_delay = 1  # seconds
         retries = 0
         while retries < max_retries: