diff --git a/ui/easydiffusion/tasks/render_images.py b/ui/easydiffusion/tasks/render_images.py index ab0917d6..bf263b3f 100644 --- a/ui/easydiffusion/tasks/render_images.py +++ b/ui/easydiffusion/tasks/render_images.py @@ -15,6 +15,7 @@ from sdkit.utils import ( img_to_base64_str, img_to_buffer, latent_samples_to_images, + log, ) from .task import Task @@ -236,7 +237,9 @@ def generate_images_internal( if convert_to_trt: pipe.unet.forward = pipe.unet._trt_forward # pipe.vae.decoder.forward = pipe.vae.decoder._trt_forward + log.info(f"Setting unet.forward to TensorRT") else: + log.info(f"Not using TensorRT for unet.forward") pipe.unet.forward = pipe.unet._non_trt_forward # pipe.vae.decoder.forward = pipe.vae.decoder._non_trt_forward setattr(pipe.unet, "_allocate_trt_buffers_backup", pipe.unet._allocate_trt_buffers)