delete unnecessary comments and messages

Signed-off-by: Rohit Jena <[email protected]>
NVIDIA · Jul 10, 2024 · 00c0fd0 · 00c0fd0
1 parent 508198f
commit 00c0fd0
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
@@ -596,8 +596,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)