delete unnecessary comments and messages

Signed-off-by: Rohit Jena <[email protected]>
rohitrango · Jul 29, 2024 · 95a8a98 · 95a8a98
1 parent 2c5e9ec
commit 95a8a98
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py b/nemo/collections/multimodal/models/text_to_image/controlnet/controlnet.py
@@ -600,8 +600,6 @@ def load_from_unet(self, from_pretrained_unet, from_NeMo=True):
                 )
                 print(f'There is {len(missing_keys)} total missing keys')
                 print("Missing:", missing_keys)
-                for key in sorted(missing_keys):
-                    print(key)
                 print("Unexpected:", unexpected_keys)
             else:
                 print("sd blocks loaded successfully")

diff --git a/nemo/collections/multimodal/modules/stable_diffusion/diffusionmodules/openaimodel.py b/nemo/collections/multimodal/modules/stable_diffusion/diffusionmodules/openaimodel.py
@@ -1219,7 +1219,7 @@ def _state_key_mapping(self, state_dict: dict):
     def _load_pretrained_model(self, state_dict, ignore_mismatched_sizes=False, from_NeMo=False):
         state_dict = self._strip_unet_key_prefix(state_dict)
         if not from_NeMo:
-            print("creating state key mapping from HF")
+            logging.info("creating state key mapping from HF")
             state_dict = self._state_key_mapping(state_dict)
         state_dict = self._legacy_unet_ckpt_mapping(state_dict)