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

adapter settings

Browse Source
This commit is contained in:
Chong
2023-08-23 16:59:56 +08:00
parent 43e9415e80
commit 8ba5c7600e
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

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

4
tests/pipelines/stable_diffusion/test_stable_diffusion_adapter.py
View File

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