Remove unnecessary flag

openvinotoolkit · Feb 22, 2024 · 539a37c · 539a37c
1 parent 5c6ffd2
commit 539a37c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/otx/algorithms/visual_prompting/tasks/openvino.py b/src/otx/algorithms/visual_prompting/tasks/openvino.py
@@ -491,15 +491,15 @@ def forward_decoder(  # type: ignore
                 )
                 has_mask_input = self.has_mask_inputs[0]
 
-            elif is_cascade and i == 1:
+            elif i == 1:
                 # Cascaded Post-refinement-1
                 mask_input, masks = self._postprocess_masks(masks, logits, scores, is_single=True)  # noqa: F821
                 if masks.sum() == 0:
                     return {"upscaled_masks": masks}
 
                 has_mask_input = self.has_mask_inputs[1]
 
-            elif is_cascade and i == 2:
+            elif i == 2:
                 # Cascaded Post-refinement-2
                 mask_input, masks = self._postprocess_masks(masks, logits, scores)  # noqa: F821
                 if masks.sum() == 0: