Update src/utils.py

src/utils.py

@@ -20,7 +20,7 @@ from diffusers.models.unet_2d_blocks import (
     get_down_block,
     get_up_block,
 )
-from diffusers.models.unet_2d_condition import UNet2DConditionOutput, logger
+from diffusers.models.unet_2d_condition import UNet2DConditionOutput
 from copy import deepcopy
 import json
 
@@ -632,7 +632,7 @@ def register_faster_forward(model, mod = '50ls'):
                 upsample_size = None
 
                 if any(s % default_overall_up_factor != 0 for s in sample.shape[-2:]):
-                    logger.info("Forward upsample size to force interpolation output size.")
+                    print("Forward upsample size to force interpolation output size.")
                     forward_upsample_size = True
 
                 # prepare attention_mask
@@ -880,7 +880,7 @@ def register_normal_forward(model):
                 # os.makedirs(f'{timestep.item()}_step', exist_ok=True)
                 #---------------------
                 if any(s % default_overall_up_factor != 0 for s in sample.shape[-2:]):
-                    logger.info("Forward upsample size to force interpolation output size.")
+                    print("Forward upsample size to force interpolation output size.")
                     forward_upsample_size = True
 
                 # prepare attention_mask