Fix vertexai (#633)

5ff60eda · Liangsheng Yin · GitHub · c1930022 · 5ff60eda
Unverified Commit 5ff60eda authored Jul 16, 2024 by Liangsheng Yin Committed by GitHub Jul 16, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 4 deletions

python/sglang/backend/vertexai.py python/sglang/backend/vertexai.py +1 -4

No files found.
--- 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):