1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00

debugging

This commit is contained in:
sayakpaul
2023-08-23 16:00:46 +05:30
parent d44c0b0182
commit a893a43231

View File

@@ -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