mirror of
https://github.com/Laurent2916/Diffusers.jl.git
synced 2024-09-18 18:45:28 +00:00
🐛 (DDPM) fix bad _extract
This commit is contained in:
parent
bb28669b29
commit
9e23327e72
|
@ -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
|
||||||
|
|
Loading…
Reference in a new issue