diff --git a/common/model.py b/common/model.py index f37c765..dc65ed7 100644 --- a/common/model.py +++ b/common/model.py @@ -13,7 +13,7 @@ from common import config from common.logger import get_loading_progress_bar from common.utils import unwrap -if not os.getenv("EXPORT_OPENAPI"): +if not os.getenv("EXPORT_OPENAPI", "").lower() in ("true", "1"): from backends.exllamav2.model import ExllamaV2Container # Global model container diff --git a/main.py b/main.py index 03f6be7..efb153f 100644 --- a/main.py +++ b/main.py @@ -29,7 +29,7 @@ async def entrypoint(args: Optional[dict] = None): signal.signal(signal.SIGINT, signal_handler) signal.signal(signal.SIGTERM, signal_handler) - if os.getenv("EXPORT_OPENAPI").lower() in ("true", "1"): + if os.getenv("EXPORT_OPENAPI", "").lower() in ("true", "1"): openapi_json = export_openapi() async with aiofiles.open("openapi.json", "w") as f: