mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-21 13:48:46 +00:00
add missing device and dtype to SD1UNet's UpBlocks
This commit is contained in:
parent
444882a734
commit
04fd98ce20
|
@ -188,7 +188,7 @@ class SD1UNet(fl.Chain):
|
|||
fl.UseContext(context="unet", key="residuals").compose(lambda x: x[-1]),
|
||||
MiddleBlock(device=device, dtype=dtype),
|
||||
),
|
||||
UpBlocks(),
|
||||
UpBlocks(device=device, dtype=dtype),
|
||||
fl.Chain(
|
||||
fl.GroupNorm(channels=320, num_groups=32, device=device, dtype=dtype),
|
||||
fl.SiLU(),
|
||||
|
|
Loading…
Reference in a new issue