From 5ff60eda7829cf075fe607b1ba2cf7be66917168 Mon Sep 17 00:00:00 2001 From: Liangsheng Yin Date: Tue, 16 Jul 2024 16:07:19 -0700 Subject: [PATCH] Fix vertexai (#633) --- python/sglang/backend/vertexai.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/python/sglang/backend/vertexai.py b/python/sglang/backend/vertexai.py index 07827b76d..871cb4f88 100644 --- a/python/sglang/backend/vertexai.py +++ b/python/sglang/backend/vertexai.py @@ -14,15 +14,12 @@ try: GenerativeModel, Image, ) - from vertexai.generative_models._generative_models import SafetySettingsType except ImportError as e: GenerativeModel = e class VertexAI(BaseBackend): - def __init__( - self, model_name, safety_settings: Optional[SafetySettingsType] = None - ): + def __init__(self, model_name, safety_settings=None): super().__init__() if isinstance(GenerativeModel, Exception):