Unverified Commit 2636d762 authored by Jee Jee Li's avatar Jee Jee Li Committed by GitHub
Browse files

[Misc] Remove unused ModelKeys (#32608)


Signed-off-by: default avatarJee Jee Li <pandaleefree@gmail.com>
parent aa7f37cc
...@@ -8,44 +8,7 @@ from dataclasses import dataclass, field ...@@ -8,44 +8,7 @@ from dataclasses import dataclass, field
@dataclass @dataclass
class ModelKeys: class MultiModelKeys:
model_type: str = None
module_list: str = None
embedding: str = None
mlp: str = None
down_proj: str = None
attention: str = None
o_proj: str = None
q_proj: str = None
k_proj: str = None
v_proj: str = None
qkv_proj: str = None
qk_proj: str = None
qa_proj: str = None
qb_proj: str = None
kva_proj: str = None
kvb_proj: str = None
output: str = None
@dataclass
class MultiModelKeys(ModelKeys):
language_model: list[str] = field(default_factory=list) language_model: list[str] = field(default_factory=list)
connector: list[str] = field(default_factory=list) connector: list[str] = field(default_factory=list)
# vision tower and audio tower # vision tower and audio tower
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment