From 6bf9928e57a08cfb17c65feeaad3ee1a129f727c Mon Sep 17 00:00:00 2001 From: Finlay Hudson Date: Wed, 14 Feb 2024 14:56:47 +0000 Subject: [PATCH] Set max image resolution to be 1280 --- .../pipelines/stable_diffusion_xl_trt/models/model_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/stable_diffusion_xl_trt/models/model_utils.py b/src/diffusers/pipelines/stable_diffusion_xl_trt/models/model_utils.py index a28942e739e8..5973131145aa 100644 --- a/src/diffusers/pipelines/stable_diffusion_xl_trt/models/model_utils.py +++ b/src/diffusers/pipelines/stable_diffusion_xl_trt/models/model_utils.py @@ -200,7 +200,7 @@ def __init__(self, self.min_batch = 1 self.max_batch = max_batch_size self.min_image_shape = 256 # min image resolution: 256x256 - self.max_image_shape = 1024 # max image resolution: 1024x1024 + self.max_image_shape = 1280 # max image resolution: 1280x1280 self.min_latent_shape = self.min_image_shape // 8 self.max_latent_shape = self.max_image_shape // 8