diff --git a/optimum/neuron/accelerate/state.py b/optimum/neuron/accelerate/state.py index 6a2b98ec5..0da4ae002 100644 --- a/optimum/neuron/accelerate/state.py +++ b/optimum/neuron/accelerate/state.py @@ -296,7 +296,6 @@ def __init__( self.distributed_type = NeuronDistributedType.MODEL_PARALLELISM self.mp_plugin = mp_plugin - print("MP PLUGIN", self.mp_plugin) if torch.distributed.is_initialized() and not parallel_state.model_parallel_is_initialized(): parallel_state.initialize_model_parallel(