Unverified Commit d2815879 authored by Enrique Shockwave's avatar Enrique Shockwave Committed by GitHub
Browse files

Improve: Support xgrammar 0.1.14 (#3593)

parent b0df5d24
...@@ -35,7 +35,7 @@ runtime_common = [ ...@@ -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 ...@@ -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): ...@@ -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
......
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