From a893a4323179eb44ffb077a4cee9a7b4df622329 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Wed, 23 Aug 2023 16:00:46 +0530 Subject: [PATCH] debugging --- src/diffusers/models/adapter.py | 2 ++ 1 file changed, 2 insertions(+) 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