Update cumo/model/language_model/llava_llama.py

cumo/model/language_model/llava_llama.py

@@ -119,9 +119,10 @@ class LlavaLlamaForCausalLM(LlamaForCausalLM, LlavaMetaForCausalLM):
                 inputs,
                 position_ids,
                 attention_mask,
-                _,
+                past_key_values,
                 inputs_embeds,
-                _
+                labels,
+                *_
             ) = self.prepare_inputs_labels_for_multimodal(
                 inputs,
                 position_ids,