compatible with new outlines (#3435)
This commit is contained in:
@@ -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__)
|
||||
|
||||
Reference in New Issue
Block a user