From 003ec9bea7980bb410a75e87a96f84cc2910cce0 Mon Sep 17 00:00:00 2001 From: SangioAI Date: Fri, 5 Sep 2025 12:18:12 +0200 Subject: [PATCH] fix: input_ids in llava_qwen.py generation --- llava/model/language_model/llava_qwen.py | 1 + 1 file changed, 1 insertion(+) diff --git a/llava/model/language_model/llava_qwen.py b/llava/model/language_model/llava_qwen.py index 70752a2..a9f7019 100644 --- a/llava/model/language_model/llava_qwen.py +++ b/llava/model/language_model/llava_qwen.py @@ -112,6 +112,7 @@ def generate( ) -> Union[GenerateOutput, torch.LongTensor]: position_ids = kwargs.pop("position_ids", None) attention_mask = kwargs.pop("attention_mask", None) + inputs = kwargs.pop("input_ids", inputs) if "inputs_embeds" in kwargs: raise NotImplementedError("`inputs_embeds` is not supported")