Fix aiohttp timeout settings (#986)

serge-chat · Dec 18, 2023 · bed108f · bed108f
1 parent a43c13f
commit bed108f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/src/serge/routers/model.py b/api/src/serge/routers/model.py
@@ -146,7 +146,7 @@ async def download_model(model_name: str):
         model_path = os.path.join(WEIGHTS, f"{model_name}.bin")
 
         # Create an aiohttp session with timeout settings
-        timeout = aiohttp.ClientTimeout(total=300)
+        timeout = aiohttp.ClientTimeout(total=None, connect=300, sock_read=300)
         async with aiohttp.ClientSession(timeout=timeout) as session:
             # Start the download and add to active_downloads
             download_task = asyncio.create_task(download_file(session, model_url, temp_model_path))