minicpm working without uhd

This commit is contained in:
Xuan Son Nguyen 2025-01-23 12:14:06 +01:00
parent c0d93dd509
commit 8586d23c8a
9 changed files with 77 additions and 2 deletions

View file

@ -907,6 +907,22 @@ class TensorNameMap:
MODEL_TENSOR.V_RESMPL_QUERY: (
"resampler.query",
),
MODEL_TENSOR.V_TOK_EMBD_IMAGE:(
"v.tok_embd.image", # tensor generated from token embeddings
),
MODEL_TENSOR.V_TOK_EMBD_END_IMAGE:(
"v.tok_embd.end_image", # tensor generated from token embeddings
),
MODEL_TENSOR.V_TOK_EMBD_SLICE:(
"v.tok_embd.slice", # tensor generated from token embeddings
),
MODEL_TENSOR.V_TOK_EMBD_END_SLICE:(
"v.tok_embd.end_slice", # tensor generated from token embeddings
),
}
# architecture-specific block mappings