ARM: Update vllm/platforms/cpu.py

Co-authored-by: Nikhil Gupta <nikhil.gupta2@arm.com>
Signed-off-by: Akash Kaothalkar <akash.kaothalkar@ibm.com>

Author: 2 people

vllm/platforms/cpu.py

@@ -266,4 +266,4 @@ def default_v1(cls, model_config) -> bool:
         """
         arch = cls.get_cpu_architecture()
         return (cls.supports_v1(model_config)
-                and arch in (CpuArchEnum.X86, CpuArchEnum.POWERPC))
+                and arch in (CpuArchEnum.X86, CpuArchEnum.POWERPC, CpuArchEnum.ARM))