From 329053892ad139509f9e24925f77493a997d9d20 Mon Sep 17 00:00:00 2001 From: Chong Date: Tue, 22 Aug 2023 11:50:41 +0800 Subject: [PATCH] update --- src/diffusers/models/adapter.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/diffusers/models/adapter.py b/src/diffusers/models/adapter.py index 04cf9d9ba1..0fd22e91c9 100644 --- a/src/diffusers/models/adapter.py +++ b/src/diffusers/models/adapter.py @@ -110,7 +110,7 @@ class T2IAdapter(ModelMixin, ConfigMixin): if adapter_type == "full_adapter": self.adapter = FullAdapter(in_channels, channels, num_res_blocks, downscale_factor) elif adapter_type == "full_adapter_xl": - self.adapter = FullAdapter_XL(in_channels, channels, num_res_blocks, downscale_factor) + self.adapter = FullAdapterXL(in_channels, channels, num_res_blocks, downscale_factor) elif adapter_type == "light_adapter": self.adapter = LightAdapter(in_channels, channels, num_res_blocks, downscale_factor) else: @@ -167,7 +167,7 @@ class FullAdapter(nn.Module): return features -class FullAdapter_XL(nn.Module): +class FullAdapterXL(nn.Module): def __init__( self, in_channels: int = 3,