diff --git a/api/onnx_web/diffusers/load.py b/api/onnx_web/diffusers/load.py index 8d006f76..366ccb17 100644 --- a/api/onnx_web/diffusers/load.py +++ b/api/onnx_web/diffusers/load.py @@ -229,10 +229,9 @@ def load_pipeline( tokenizer_2=components.get("tokenizer_2", None), ) else: - logger.debug("assembling SD pipeline for %s", pipeline_class.__name__) - - if pipeline_class == OnnxStableDiffusionUpscalePipeline: + if "vae" in components: # upscale uses a single VAE + logger.debug("assembling SD pipeline for %s with single VAE", pipeline_class.__name__) pipe = pipeline_class( components["vae"], components["text_encoder"], @@ -242,6 +241,7 @@ def load_pipeline( scheduler, ) else: + logger.debug("assembling SD pipeline for %s with VAE codec", pipeline_class.__name__) pipe = pipeline_class( components["vae_encoder"], components["vae_decoder"], diff --git a/api/onnx_web/diffusers/pipelines/panorama.py b/api/onnx_web/diffusers/pipelines/panorama.py index 810cf0e7..317a8515 100644 --- a/api/onnx_web/diffusers/pipelines/panorama.py +++ b/api/onnx_web/diffusers/pipelines/panorama.py @@ -563,6 +563,8 @@ class OnnxStableDiffusionPanoramaPipeline(DiffusionPipeline): # panorama additions views, resize = self.get_views(height, width, self.window, self.stride) + logger.trace("panorama resized latents to %s", resize) + count = np.zeros(resize_latent_shape(latents, resize)) value = np.zeros(resize_latent_shape(latents, resize)) @@ -977,6 +979,8 @@ class OnnxStableDiffusionPanoramaPipeline(DiffusionPipeline): # panorama additions views, resize = self.get_views(height, width, self.window, self.stride) + logger.trace("panorama resized latents to %s", resize) + count = np.zeros(resize_latent_shape(latents, resize)) value = np.zeros(resize_latent_shape(latents, resize)) @@ -1298,6 +1302,8 @@ class OnnxStableDiffusionPanoramaPipeline(DiffusionPipeline): # panorama additions views, resize = self.get_views(height, width, self.window, self.stride) + logger.trace("panorama resized latents to %s", resize) + count = np.zeros(resize_latent_shape(latents, resize)) value = np.zeros(resize_latent_shape(latents, resize)) diff --git a/api/onnx_web/diffusers/pipelines/panorama_xl.py b/api/onnx_web/diffusers/pipelines/panorama_xl.py index 2650a2a3..5551267b 100644 --- a/api/onnx_web/diffusers/pipelines/panorama_xl.py +++ b/api/onnx_web/diffusers/pipelines/panorama_xl.py @@ -394,6 +394,8 @@ class StableDiffusionXLPanoramaPipelineMixin(StableDiffusionXLImg2ImgPipelineMix # 8. Panorama additions views, resize = self.get_views(height, width, self.window, self.stride) + logger.trace("panorama resized latents to %s", resize) + count = np.zeros(resize_latent_shape(latents, resize)) value = np.zeros(resize_latent_shape(latents, resize)) @@ -819,6 +821,8 @@ class StableDiffusionXLPanoramaPipelineMixin(StableDiffusionXLImg2ImgPipelineMix # 8. Panorama additions views, resize = self.get_views(height, width, self.window, self.stride) + logger.trace("panorama resized latents to %s", resize) + count = np.zeros(resize_latent_shape(latents, resize)) value = np.zeros(resize_latent_shape(latents, resize))