diff --git a/library/train_util.py b/library/train_util.py index 6aa0631e..dc5adee5 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -5710,7 +5710,6 @@ def sample_image_inference( controlnet=controlnet, controlnet_image=controlnet_image, ) - logger.info(f"latents: {latents.shape}") clean_memory_on_device(accelerator.device) image = pipeline.latents_to_image(latents)[0] @@ -5720,7 +5719,6 @@ def sample_image_inference( clean_memory_on_device(accelerator.device) original_latent = prompt_dict.get("original_lantent").to(device=accelerator.device) - logger.info(f"original_latent: {original_latent.shape}") original_image = pipeline.latents_to_image(original_latent)[0] text_image = draw_text_on_image(f"caption: {prompt}", image.width * 2) new_image = Image.new('RGB', (original_image.width + image.width, original_image.height + text_image.height))