Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-23 14:48:45 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add missing device and dtype to SD1UNet's UpBlocks

Browse source
This commit is contained in:
Laurent 2024-09-08 13:47:35 +00:00 committed by Laureηt
parent 444882a734
commit 602c63ce76
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/refiners/foundationals/latent_diffusion/stable_diffusion_1/unet.py
View file

@ -188,7 +188,7 @@ class SD1UNet(fl.Chain):
fl.UseContext(context="unet", key="residuals").compose(lambda x: x[-1]), fl.UseContext(context="unet", key="residuals").compose(lambda x: x[-1]),
MiddleBlock(device=device, dtype=dtype), MiddleBlock(device=device, dtype=dtype),
), ),
UpBlocks(), UpBlocks(device=device, dtype=dtype),
fl.Chain( fl.Chain(
fl.GroupNorm(channels=320, num_groups=32, device=device, dtype=dtype), fl.GroupNorm(channels=320, num_groups=32, device=device, dtype=dtype),
fl.SiLU(), fl.SiLU(),