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 = [
"torchao>=0.7.0",
"uvicorn",
"uvloop",
"xgrammar==0.1.10",
"xgrammar==0.1.14",
"ninja",
"transformers==4.48.3",
"llguidance>=0.6.15"
......
......@@ -19,7 +19,6 @@ from typing import List, Tuple
import torch
from xgrammar import (
CompiledGrammar,
Grammar,
GrammarCompiler,
GrammarMatcher,
TokenizerInfo,
......@@ -135,9 +134,7 @@ class XGrammarGrammarBackend(BaseGrammarBackend):
return None
elif key_type == "regex":
try:
ctx = self.grammar_compiler.compile_grammar(
Grammar.from_regex(key_string)
)
ctx = self.grammar_compiler.compile_regex(key_string)
except RuntimeError as e:
logging.warning(f"Skip invalid regex: regex={key_string}, {e=}")
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