diff --git a/src/diffusers/models/adapter.py b/src/diffusers/models/adapter.py index fabe5fd94e..5b0d0f8793 100644 --- a/src/diffusers/models/adapter.py +++ b/src/diffusers/models/adapter.py @@ -136,6 +136,7 @@ class FullAdapter(nn.Module): downscale_factor: int = 8, ): super().__init__() + print(f"From {self.__class__.name} channels: {channels}.") in_channels = in_channels * downscale_factor**2 @@ -265,6 +266,7 @@ class LightAdapter(nn.Module): downscale_factor: int = 8, ): super().__init__() + print(f"From {self.__class__.name} channels: {channels}.") in_channels = in_channels * downscale_factor**2