diff --git a/python/sglang/srt/model_executor/model_runner.py b/python/sglang/srt/model_executor/model_runner.py index d06b28124..ca67da75e 100644 --- a/python/sglang/srt/model_executor/model_runner.py +++ b/python/sglang/srt/model_executor/model_runner.py @@ -1044,10 +1044,6 @@ class ModelRunner: self.server_args.load_format = load_format self.load_config = load_config - # Recapture device graph after model weight update. - if not self.server_args.disable_cuda_graph and self.device == "cuda": - self.init_device_graphs() - logger.info("Update weights end.") return True, "Succeeded to update model weights."