Refactor resolve_beam to fix recursion depth issue

Signed-off-by: Zhang, Yanli L <[email protected]>

optimum/habana/transformers/generation/utils.py

@@ -2764,12 +2764,17 @@ def finalize_beams(initial_ids, beam_trace, model_config, length_penalty):
             root = (float("-inf"), None, None, False)
 
             def resolve_beam(beam):
-                if beam == root:
-                    return []
-                score, prev, tok, is_finished = beam
-                rest = resolve_beam(prev)
-                rest.append(tok)
-                return rest
+                tokens = []
+                current = beam
+
+                while current != root:
+                    score, prev, tok, is_finished = current
+                    tokens.append(tok)
+                    current = prev
+
+                # Reverse tokens since we are adding them in reverse order
+                tokens.reverse()
+                return tokens
 
             prev_beams = [[root] * num_beams] * bs
             best = [root] * bs