compatible with new outlines (#3435)

2025-02-10 01:51:30 +08:00
parent 64c8713573
commit 85986bb978
2 changed files with 6 additions and 3 deletions
--- a/python/sglang/srt/constrained/outlines_backend.py
+++ b/python/sglang/srt/constrained/outlines_backend.py
@@ -35,7 +35,10 @@ is_hip_ = is_hip()
 if is_hip_:
    from outlines_core.fsm.json_schema import build_regex_from_schema
 else:
-    from outlines.fsm.json_schema import build_regex_from_schema
+    try:
+        from outlines.fsm.json_schema import build_regex_from_schema
+    except ImportError:
+        from outlines_core.fsm.json_schema import build_regex_from_schema


 logger = logging.getLogger(__name__)