add log

Author: lyogavin

examples/text_to_image/train_img2img_test.py

@@ -464,8 +464,8 @@ def main():
         args.pretrained_model_name_or_path, subfolder="unet", in_channels=8, revision=args.non_ema_revision,
                                                 ignore_mismatched_sizes=True,
     )
-    print(unet)
-    print(unet.conv_in)
+    #print(unet)
+    #print(unet.conv_in)
 
     # Freeze vae and text_encoder
     vae.requires_grad_(False)
@@ -477,8 +477,8 @@ def main():
             args.pretrained_model_name_or_path, subfolder="unet", in_channels=8, revision=args.revision,
                                                 ignore_mismatched_sizes=True,
         )
-        print(ema_unet)
-        print(unet.conv_in)
+        #print(ema_unet)
+        #print(unet.conv_in)
         ema_unet = EMAModel(ema_unet.parameters())
 
     if args.enable_xformers_memory_efficient_attention: