refactor: move class_pos_emb to VAETKI case

This commit is contained in:
suhyun-hwang 2026-01-10 20:44:42 +09:00
parent c5e9eac8c5
commit 9d531ea9d5
1 changed files with 1 additions and 1 deletions

View File

@ -1336,7 +1336,6 @@ struct clip_model_loader {
};
model.class_embedding = get_tensor(TN_CLASS_EMBD, false);
model.class_pos_emb = get_tensor(TN_CLASS_POS_EMBD, false);
model.pre_ln_w = get_tensor(string_format(TN_LN_PRE, prefix, "weight"), false);
model.pre_ln_b = get_tensor(string_format(TN_LN_PRE, prefix, "bias"), false);
@ -1554,6 +1553,7 @@ struct clip_model_loader {
} break;
case PROJECTOR_TYPE_VAETKI:
{
model.class_pos_emb = get_tensor(TN_CLASS_POS_EMBD);
model.mm_0_w = get_tensor(string_format(TN_MVLM_PROJ_MLP, 0, "weight"));
model.mm_0_b = get_tensor(string_format(TN_MVLM_PROJ_MLP, 0, "bias"));
model.mm_1_w = get_tensor(string_format(TN_MVLM_PROJ_MLP, 1, "weight"));