diff --git a/deepspeed/inference/v2/model_implementations/common_parameters/embedding_parameters.py b/deepspeed/inference/v2/model_implementations/common_parameters/embedding_parameters.py index 55f9999a47cf..2ed34b5fd259 100644 --- a/deepspeed/inference/v2/model_implementations/common_parameters/embedding_parameters.py +++ b/deepspeed/inference/v2/model_implementations/common_parameters/embedding_parameters.py @@ -23,5 +23,4 @@ class EmbeddingParameter(ParameterBase): """ def finalize(self) -> torch.Tensor: - print("EmbeddingParameter.finalize") return self.inference_model.transform_embedding_param(self.params)