diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index 58814699b3..a52ab359bd 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -283,6 +283,9 @@ def unsloth_base_fast_generate(self, *args, **kwargs): input_ids = kwargs["input"] elif "input_features" in kwargs: input_ids = kwargs["input_features"] + elif "inputs_embeds" in kwargs: + # canonical HF name for embedding inputs (e.g. multimodal generate) + input_ids = kwargs["inputs_embeds"] elif "input_embeds" in kwargs: input_ids = kwargs["input_embeds"] elif "inputs" in kwargs: