[Iluvatar][CI] fix safetensors_rust.SafetensorError: framework paddle is invalid (#4972)

This commit is contained in:
yzwu
2025-11-12 14:13:40 +08:00
committed by GitHub
parent 35bd2afab3
commit 76e60e98f8
3 changed files with 3 additions and 2 deletions
+1 -1
View File
@@ -38,7 +38,7 @@ def get_moe_method():
return moe method based on device platform
"""
if current_platform.is_cuda():
if current_platform.is_cuda() or current_platform.is_iluvatar():
from .fused_moe_cutlass_backend import CutlassMoEMethod
return CutlassMoEMethod(None)