From be3065260f19b02a807c93b572a6d10ab396bd57 Mon Sep 17 00:00:00 2001 From: Bingxin Ke Date: Sun, 28 Apr 2024 15:45:40 +0200 Subject: [PATCH] [FIX] type error when passing ensemble res --- marigold/util/ensemble.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/marigold/util/ensemble.py b/marigold/util/ensemble.py index b4dc0d2..5a2908e 100644 --- a/marigold/util/ensemble.py +++ b/marigold/util/ensemble.py @@ -61,7 +61,7 @@ def ensemble_depths( scale_factor = torch.min(max_res / torch.tensor(ori_shape[-2:])) if scale_factor < 1: downscaler = torch.nn.Upsample(scale_factor=scale_factor, mode="nearest") - input_images = downscaler(torch.from_numpy(input_images)).numpy() + input_images = downscaler(input_images) # init guess _min = np.min(input_images.reshape((n_img, -1)).cpu().numpy(), axis=1)