From 9e23327e725983a2e6ecb4a970e371f1e8369bb7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Laure=CE=B7t?= Date: Mon, 14 Aug 2023 18:12:37 +0200 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20(DDPM)=20fix=20bad=20=5Fextract?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/Schedulers/DDPM.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/Schedulers/DDPM.jl b/src/Schedulers/DDPM.jl index e78d40d..abd7201 100644 --- a/src/Schedulers/DDPM.jl +++ b/src/Schedulers/DDPM.jl @@ -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