Remove print

huggingface · Sep 21, 2023 · a2d349c · a2d349c
1 parent 162fcbf
commit a2d349c
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/optimum/neuron/distributed/base.py b/optimum/neuron/distributed/base.py
@@ -207,11 +207,6 @@ def parallelize(
                 # This module has not pre-trained weights, it must be fine-tuned, we initialize it with the
                 # `reset_parameters()` method.
                 mod.reset_parameters()
-            for name, mod in model.named_parameters():
-                if name in weight_map:
-                    weight_info = WeightInformation(weight_map[name], name, device=device)
-                    tensor = load_tensor_for_weight(weight_info)
-                    print(mod, tensor)
         return model
 
     @classmethod