Update registry.py

Added Isaac model architecture.

Signed-off-by: oscardev256 <42308241+oscardev256@users.noreply.github.com>
This commit is contained in:
oscardev256 2025-11-08 23:15:17 -05:00 committed by Yang Liu
parent f053784631
commit e27cb3c53d

View File

@ -326,6 +326,7 @@ _MULTIMODAL_MODELS = {
"idefics3",
"Idefics3ForConditionalGeneration",
),
"IsaacForConditionalGeneration": ("isaac", "IsaacForConditionalGeneration"),
"SmolVLMForConditionalGeneration": ("smolvlm", "SmolVLMForConditionalGeneration"), # noqa: E501
"KeyeForConditionalGeneration": ("keye", "KeyeForConditionalGeneration"),
"KeyeVL1_5ForConditionalGeneration": (