From 8ba5c7600e69c60d4dccc5842563a38fbdcbb3e9 Mon Sep 17 00:00:00 2001 From: Chong Date: Wed, 23 Aug 2023 16:59:56 +0800 Subject: [PATCH] adapter settings --- src/diffusers/models/adapter.py | 2 +- .../stable_diffusion/test_stable_diffusion_adapter.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/diffusers/models/adapter.py b/src/diffusers/models/adapter.py index cfeeed6ddb..fabe5fd94e 100644 --- a/src/diffusers/models/adapter.py +++ b/src/diffusers/models/adapter.py @@ -260,7 +260,7 @@ class LightAdapter(nn.Module): def __init__( self, in_channels: int = 3, - channels: List[int] = [320, 640, 1280], + channels: List[int] = [320, 640, 1280, 1280], num_res_blocks: int = 4, downscale_factor: int = 8, ): diff --git a/tests/pipelines/stable_diffusion/test_stable_diffusion_adapter.py b/tests/pipelines/stable_diffusion/test_stable_diffusion_adapter.py index 0c1dd1cfe8..a8ee1a7819 100644 --- a/tests/pipelines/stable_diffusion/test_stable_diffusion_adapter.py +++ b/tests/pipelines/stable_diffusion/test_stable_diffusion_adapter.py @@ -84,9 +84,9 @@ class AdapterTests: torch.manual_seed(0) adapter = T2IAdapter( in_channels=3, - channels=[32, 64], + channels=[320, 640, 1280, 1280], num_res_blocks=2, - downscale_factor=2, + downscale_factor=8, adapter_type=adapter_type, )