From 3a8edc156de6ab38c4899450e566eec1f0a23ffa Mon Sep 17 00:00:00 2001 From: ViperEkura <3081035982@qq.com> Date: Fri, 27 Feb 2026 19:46:00 +0800 Subject: [PATCH] =?UTF-8?q?refactor:=20=E6=9B=B4=E6=96=B0dump=5Ffiles?= =?UTF-8?q?=E5=87=BD=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- modules/utils.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/modules/utils.py b/modules/utils.py index a6ecad0..6cbc3c7 100644 --- a/modules/utils.py +++ b/modules/utils.py @@ -129,13 +129,15 @@ def dump_files( list_tensor = [arrow[key] for arrow in arrows] package[key] = list_tensor - output_package: Dict[str, Tensor] = {} + output_package: Dict[str, List[Tensor]] = {} for key in output_keys: if packing_size > 0: print(f"Packaging key: '{key}'") - package[key] = pack_sequences(package[key], packing_size, pad_value) - sequence = torch.cat(package[key]) + sequence = pack_sequences(package[key], packing_size, pad_value) + else: + sequence = package[key] + output_package[key] = sequence save_h5(out_file_path, output_package)