Unverified Commit 85986bb9 authored by Yineng Zhang's avatar Yineng Zhang Committed by GitHub
Browse files

compatible with new outlines (#3435)

parent 64c87135
......@@ -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
......
......@@ -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__)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment