Remove print that should not be there (#552)

huggingface · Apr 5, 2024 · 326d79b · 326d79b
1 parent 6856557
commit 326d79b
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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(