compatible with new outlines (#3435)

This commit is contained in:
Yineng Zhang
2025-02-10 01:51:30 +08:00
committed by GitHub
parent 64c8713573
commit 85986bb978
2 changed files with 6 additions and 3 deletions

View File

@@ -25,8 +25,8 @@ runtime_common = [
]
srt = [
"sglang[runtime_common]", "cuda-python",
"sgl-kernel>=0.0.3.post2", "torch", "vllm==0.6.4.post1",
"flashinfer_python>=0.2.0.post2", "outlines>=0.0.44,<0.1.0"
"sgl-kernel>=0.0.3.post2", "torch", "vllm>=0.6.4.post1,<=0.7.2",
"flashinfer_python>=0.2.0.post2", "outlines>=0.0.44,<=0.1.11"
]
# HIP (Heterogeneous-computing Interface for Portability) for AMD

View File

@@ -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__)