From fbf0b9a6e20e2f58f8fe967a98a84bb08a4c7cd8 Mon Sep 17 00:00:00 2001 From: Michael Benayoun Date: Tue, 17 Sep 2024 15:25:13 +0200 Subject: [PATCH] Comment print --- optimum/neuron/distributed/base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/optimum/neuron/distributed/base.py b/optimum/neuron/distributed/base.py index f3733fa91..98d211be3 100644 --- a/optimum/neuron/distributed/base.py +++ b/optimum/neuron/distributed/base.py @@ -777,8 +777,8 @@ def should_parallelize_layer_predicate_func(layer): f"Could not find information for the parameter {name} to set its `requires_grad` attribute." ) - for name, parameter in model.named_parameters(): - print(f"{name} => {parameter.requires_grad}") + # for name, parameter in model.named_parameters(): + # print(f"{name} => {parameter.requires_grad}") if is_main_worker(): logger.info("Load and initialization of the weights done.")