From ff2ca158446ea28a39e6fbac2e7e183928e19386 Mon Sep 17 00:00:00 2001 From: Ingvar Date: Mon, 5 Oct 2026 12:02:17 +0300 Subject: [PATCH] fix empty relex data item batch --- gliner/data_processing/processor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gliner/data_processing/processor.py b/gliner/data_processing/processor.py index cd8367d2..27e856c6 100644 --- a/gliner/data_processing/processor.py +++ b/gliner/data_processing/processor.py @@ -2022,7 +2022,7 @@ def create_relation_labels( if "span_label" in batch and batch["span_label"] is not None: batch_ents = (batch["span_label"] > 0).sum(-1) else: - batch_ents = span_mask.long().squeeze(-1).sum(-1) + batch_ents = span_mask.long().sum(-1) # Batch CPU transfer to avoid per-element .item() sync batch_ents_cpu = batch_ents.tolist()