github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add max_length=3 to the LLaVA OneVision image input field.

Browse Source
This commit is contained in:
Ryan Dick
2025-03-13 13:34:42 +00:00
committed by psychedelicious
parent 2a09604baf
commit e6dd721e33
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/app/invocations/llava_onevision_vllm.py
View File

@@ -14,7 +14,7 @@ from invokeai.backend.util.devices import TorchDevice
class LlavaOnevisionVllmInvocation(BaseInvocation):
"""Run a LLaVA OneVision VLLM model."""
images: list[ImageField] | ImageField | None = InputField(default=None, description="Input image.")
images: list[ImageField] | ImageField | None = InputField(default=None, max_length=3, description="Input image.")
prompt: str = InputField(
default="",
description="Input text prompt.",