From e0404ebf5ab239842ef18550e57c91b2c2d7e281 Mon Sep 17 00:00:00 2001
From: Fam Zheng <fam@euphon.net>
Date: Sat, 27 Dec 2025 11:04:15 +0000
Subject: [PATCH] common.py: restore results_per_img = ncells * ncells * 2

---
 emblem5/ai/common.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/emblem5/ai/common.py b/emblem5/ai/common.py
index f5e26e0..3d4c036 100644
--- a/emblem5/ai/common.py
+++ b/emblem5/ai/common.py
@@ -150,7 +150,7 @@ def make_stripe_img(left, right, nstripes):
     return ret
 
 def predict_multi(model, transforms, images, ncells=1):
-    results_per_img = ncells * ncells
+    results_per_img = ncells * ncells * 2
     ret = []
     with torch.no_grad():
         tensors = []