server: fixes initial timeout for llm warmup

This commit is contained in:
Mathieu Virbel
2023-08-11 15:45:28 +02:00
parent 38a5ee0da2
commit a06056f9bc
2 changed files with 2 additions and 2 deletions

View File

@@ -35,7 +35,7 @@ class LLM:
name = self.__class__.__name__
logger.info(f"LLM[{name}] warming up...")
try:
await retry(self._warmup)(logger=logger)
await self._warmup(logger=logger)
duration = monotonic() - start
logger.info(f"LLM[{name}] warmup took {duration:.2f} seconds")
except Exception:

View File

@@ -19,7 +19,7 @@ class ModalLLM(LLM):
response = await client.post(
self.llm_warmup_url,
headers=self.headers,
timeout=self.timeout,
timeout=60**5,
)
response.raise_for_status()