[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)
+1
View File
@@ -39,3 +39,4 @@ opentelemetry-exporter-otlp
opentelemetry-instrumentation-fastapi
partial_json_parser
msgspec
safetensors==0.7.0rc0
+1 -1
View File
@@ -42,7 +42,7 @@ def timeout(seconds):
return decorator
@timeout(150)
@timeout(180)
def offline_infer_check():
set_random_seed(123)