Laurent/Diffusers.jl
1
0
Fork 0
mirror of https://github.com/Laurent2916/Diffusers.jl.git synced 2024-09-16 17:45:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

🐛 (DDPM) fix bad _extract

Browse source
This commit is contained in:
Laureηt 2023-08-14 18:12:37 +02:00
parent bb28669b29
commit 9e23327e72
Signed by: Laurent
SSH key fingerprint: SHA256:kZEpW8cMJ54PDeCvOhzreNr4FSh6R13CMGH/POoO8DI
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/Schedulers/DDPM.jl
View file

@ -8,7 +8,7 @@ function _extract(
) )
new_size = tuple( new_size = tuple(
fill(1, ndims(reference) - 1)..., fill(1, ndims(reference) - 1)...,
size(reference, ndims(reference)) size(target, 1)
) )
return reshape(target, new_size) return reshape(target, new_size)
end end