Comment print

huggingface · Sep 17, 2024 · fbf0b9a · fbf0b9a
1 parent 013cd6a
commit fbf0b9a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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.")