[Misc] Remove unused ModelKeys (#32608)
Signed-off-by: Jee Jee Li <pandaleefree@gmail.com>
This commit is contained in:
@@ -8,44 +8,7 @@ from dataclasses import dataclass, field
|
||||
|
||||
|
||||
@dataclass
|
||||
class ModelKeys:
|
||||
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):
|
||||
class MultiModelKeys:
|
||||
language_model: list[str] = field(default_factory=list)
|
||||
connector: list[str] = field(default_factory=list)
|
||||
# vision tower and audio tower
|
||||
|
||||
Reference in New Issue
Block a user