From b6f8069b365af95bb3f704eadcfd77043adea4d3 Mon Sep 17 00:00:00 2001 From: Yonghua Li <39643373+liyonghua0910@users.noreply.github.com> Date: Thu, 4 Dec 2025 19:43:01 +0800 Subject: [PATCH] [fix] update check_model_weights_status loop (#5249) --- fastdeploy/rl/dynamic_weight_manager.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/fastdeploy/rl/dynamic_weight_manager.py b/fastdeploy/rl/dynamic_weight_manager.py index dba4d0f27..a6b611516 100644 --- a/fastdeploy/rl/dynamic_weight_manager.py +++ b/fastdeploy/rl/dynamic_weight_manager.py @@ -258,19 +258,18 @@ class DynamicWeightManager: check model weights status """ logger.info(f"dynamic weight manager is check model weights status! {model_weights_status.value[0]}") - while model_weights_status.value[0] != ModelWeightsStatus.NORMAL: + while ( + model_weights_status.value[0] != ModelWeightsStatus.NORMAL + and model_weights_status.value[0] != ModelWeightsStatus.CLEARED + ): if model_weights_status.value[0] == ModelWeightsStatus.UPDATING: logger.info("infer engine stopped! start to load new checkpoint...") model_runner.clear_requests() model_runner.update_parameters(pid) - while model_weights_status.value[0] != ModelWeightsStatus.NORMAL: - time.sleep(0.01) logger.info("finished loading new checkpoint") elif model_weights_status.value[0] == ModelWeightsStatus.CLEARING: logger.info("infer engine stopped! start to clear checkpoint...") model_runner.clear_requests() model_runner.clear_parameters(pid) - while model_weights_status.value[0] != ModelWeightsStatus.CLEARED: - time.sleep(0.01) logger.info("finished clearing checkpoint") time.sleep(0.01)