mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2026-04-23 00:17:25 +08:00
[Iluvatar][CI] fix safetensors_rust.SafetensorError: framework paddle is invalid (#4972)
This commit is contained in:
@@ -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)
|
||||
|
||||
@@ -39,3 +39,4 @@ opentelemetry-exporter-otlp
|
||||
opentelemetry-instrumentation-fastapi
|
||||
partial_json_parser
|
||||
msgspec
|
||||
safetensors==0.7.0rc0
|
||||
|
||||
@@ -42,7 +42,7 @@ def timeout(seconds):
|
||||
return decorator
|
||||
|
||||
|
||||
@timeout(150)
|
||||
@timeout(180)
|
||||
def offline_infer_check():
|
||||
set_random_seed(123)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user