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

Merge branch 'main' of https://github.com/MC-E/diffusers

Browse Source
This commit is contained in:
Chong
2023-08-23 19:35:56 +08:00
parent b3d6a5eda8 52bc39c997
commit 732cb3e2b5
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/diffusers/models/adapter.py
View File

@@ -163,7 +163,7 @@ class FullAdapter(nn.Module):
for block in self.body:
x = block(x)
features.append(x)
return features
@@ -265,6 +265,7 @@ class LightAdapter(nn.Module):
downscale_factor: int = 8,
):
super().__init__()
print(f"From {self.__class__} channels: {channels}.")
in_channels = in_channels * downscale_factor**2
@@ -291,7 +292,7 @@ class LightAdapter(nn.Module):
for block in self.body:
x = block(x)
features.append(x)
return features