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

update

Browse Source
This commit is contained in:
Chong
2023-08-22 11:50:41 +08:00
parent a554f5a7ee
commit 329053892a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

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