From 2332afbac80215f91b50f15f2384f4acba2e8059 Mon Sep 17 00:00:00 2001 From: Vidya Galli Date: Tue, 1 Oct 2024 07:49:22 -0700 Subject: [PATCH] Fix local variable 'image_features' referenced before assignment (#1383) --- optimum/habana/transformers/models/llava/modeling_llava.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/optimum/habana/transformers/models/llava/modeling_llava.py b/optimum/habana/transformers/models/llava/modeling_llava.py index cccbf8ebb9..402a1850fe 100644 --- a/optimum/habana/transformers/models/llava/modeling_llava.py +++ b/optimum/habana/transformers/models/llava/modeling_llava.py @@ -153,7 +153,8 @@ def forward( # 1. Extra the input embeddings inputs_embeds = self.get_input_embeddings()(input_ids) - + + image_features = None # 2. Merge text and images if pixel_values is not None and input_ids.shape[1] != 1: image_outputs = self.vision_tower(