diff --git a/src/engine.py b/src/engine.py index 80bf72b..75f007e 100644 --- a/src/engine.py +++ b/src/engine.py @@ -358,7 +358,10 @@ class OpenAIvLLMEngine(vLLMEngine): ) if hasattr(self.chat_engine, 'warmup'): - await self.chat_engine.warmup() + import asyncio + result = self.chat_engine.warmup() + if asyncio.iscoroutine(result): + await result async def generate(self, openai_request: JobInput): # Ensure engines are ready (no-op if already initialized at startup)