From bd2b9b62c7f205301f555042ee70866227dea6b9 Mon Sep 17 00:00:00 2001 From: Chris Rosewarne Date: Thu, 20 Aug 2026 10:18:29 +1000 Subject: [PATCH] [fix] Prevent crash on image encode_only from decoded hash check --- .../pipelines/split_inference/image_split_inference.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/compressai_vision/pipelines/split_inference/image_split_inference.py b/compressai_vision/pipelines/split_inference/image_split_inference.py index 32fc25b..c1c9e7e 100644 --- a/compressai_vision/pipelines/split_inference/image_split_inference.py +++ b/compressai_vision/pipelines/split_inference/image_split_inference.py @@ -99,6 +99,7 @@ def __call__( self.init_complexity_measure() accum_enc_by_module = None accum_dec_by_module = None + dec_features = None for e, d in enumerate(tqdm(dataloader)): features = {"org_input_size": [], "input_size": []} @@ -300,7 +301,8 @@ def __call__( del features if ( - "decoded_feature_tensor_hash" in dec_features + dec_features is not None + and "decoded_feature_tensor_hash" in dec_features and dec_features["decoded_feature_tensor_hash"] is not None and self.configs["nn_task_part2"].dump_features_hash ):