diff --git a/fastdeploy/rl/dynamic_weight_manager.py b/fastdeploy/rl/dynamic_weight_manager.py index d2d75d50f..5583108dd 100644 --- a/fastdeploy/rl/dynamic_weight_manager.py +++ b/fastdeploy/rl/dynamic_weight_manager.py @@ -109,7 +109,7 @@ class DynamicWeightManager: ) try: - ipc_state_dict = paddle.load(model_path) + ipc_state_dict = paddle.load(model_path, safetensors=True) except FileNotFoundError: fallback_path = f"/shared_ipc_meta/model_state.tp0{self.meta_src_id}.pdparams" ipc_state_dict = paddle.load(fallback_path)