Improve: Support xgrammar 0.1.14 (#3593)
This commit is contained in:
committed by
GitHub
parent
b0df5d240b
commit
d281587989
@@ -35,7 +35,7 @@ runtime_common = [
|
|||||||
"torchao>=0.7.0",
|
"torchao>=0.7.0",
|
||||||
"uvicorn",
|
"uvicorn",
|
||||||
"uvloop",
|
"uvloop",
|
||||||
"xgrammar==0.1.10",
|
"xgrammar==0.1.14",
|
||||||
"ninja",
|
"ninja",
|
||||||
"transformers==4.48.3",
|
"transformers==4.48.3",
|
||||||
"llguidance>=0.6.15"
|
"llguidance>=0.6.15"
|
||||||
|
|||||||
@@ -19,7 +19,6 @@ from typing import List, Tuple
|
|||||||
import torch
|
import torch
|
||||||
from xgrammar import (
|
from xgrammar import (
|
||||||
CompiledGrammar,
|
CompiledGrammar,
|
||||||
Grammar,
|
|
||||||
GrammarCompiler,
|
GrammarCompiler,
|
||||||
GrammarMatcher,
|
GrammarMatcher,
|
||||||
TokenizerInfo,
|
TokenizerInfo,
|
||||||
@@ -135,9 +134,7 @@ class XGrammarGrammarBackend(BaseGrammarBackend):
|
|||||||
return None
|
return None
|
||||||
elif key_type == "regex":
|
elif key_type == "regex":
|
||||||
try:
|
try:
|
||||||
ctx = self.grammar_compiler.compile_grammar(
|
ctx = self.grammar_compiler.compile_regex(key_string)
|
||||||
Grammar.from_regex(key_string)
|
|
||||||
)
|
|
||||||
except RuntimeError as e:
|
except RuntimeError as e:
|
||||||
logging.warning(f"Skip invalid regex: regex={key_string}, {e=}")
|
logging.warning(f"Skip invalid regex: regex={key_string}, {e=}")
|
||||||
return None
|
return None
|
||||||
|
|||||||
Reference in New Issue
Block a user