[Bugfix] Only require XGrammar on x86 (#10865)

Signed-off-by: mgoin <michael@neuralmagic.com>
2024-12-03 13:32:21 -05:00
parent 2f2cdc745a
commit 7090c27bb2
4 changed files with 36 additions and 3 deletions
--- a/vllm/platforms/init.py
+++ b/vllm/platforms/init.py
@@ -1,5 +1,5 @@
 from .interface import _Backend  # noqa: F401
-from .interface import Platform, PlatformEnum, UnspecifiedPlatform
+from .interface import CpuArchEnum, Platform, PlatformEnum, UnspecifiedPlatform

 current_platform: Platform

@@ -120,4 +120,4 @@ elif is_openvino:
 else:
    current_platform = UnspecifiedPlatform()

-__all__ = ['Platform', 'PlatformEnum', 'current_platform']
+__all__ = ['Platform', 'PlatformEnum', 'current_platform', 'CpuArchEnum']