mirror of
https://github.com/Laurent2916/Diffusers.jl.git
synced 2024-09-16 17:45:29 +00:00
🐛 (DDPM) fix bad _extract
This commit is contained in:
parent
bb28669b29
commit
9e23327e72
|
@ -8,7 +8,7 @@ function _extract(
|
|||
)
|
||||
new_size = tuple(
|
||||
fill(1, ndims(reference) - 1)...,
|
||||
size(reference, ndims(reference))
|
||||
size(target, 1)
|
||||
)
|
||||
return reshape(target, new_size)
|
||||
end
|
||||
|
|
Loading…
Reference in a new issue