Fix IdealSizeInvocation (#6145)

This commit is contained in:
Jonathan 2024-04-04 16:38:40 -05:00 committed by GitHub
parent d284e0567a
commit 3659219f46
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1254,7 +1254,7 @@ class IdealSizeInvocation(BaseInvocation):
return tuple((x - x % multiple_of) for x in args)
def invoke(self, context: InvocationContext) -> IdealSizeOutput:
unet_config = context.models.get_config(**self.unet.unet.model_dump())
unet_config = context.models.get_config(self.unet.unet.key)
aspect = self.width / self.height
dimension: float = 512
if unet_config.base == BaseModelType.StableDiffusion2: