[VLM][Model] Support image input for Chameleon (#6633)

This commit is contained in:
Roger Wang
2024-07-22 23:50:48 -07:00
committed by GitHub
parent c5201240a4
commit 22fa2e35cb
7 changed files with 696 additions and 58 deletions

View File

@@ -1,4 +1,5 @@
from vllm.transformers_utils.configs.chameleon import ChameleonConfig
from vllm.transformers_utils.configs.chameleon import (ChameleonConfig,
ChameleonVQVAEConfig)
from vllm.transformers_utils.configs.chatglm import ChatGLMConfig
from vllm.transformers_utils.configs.dbrx import DbrxConfig
# RWConfig is for the original tiiuae/falcon-40b(-instruct) and
@@ -12,6 +13,7 @@ from vllm.transformers_utils.configs.mpt import MPTConfig
__all__ = [
"ChameleonConfig",
"ChameleonVQVAEConfig",
"ChatGLMConfig",
"DbrxConfig",
"MPTConfig",