feat: support packed INT4 convrot models (#2276)

Co-authored-by: noctrex <noctrex@gmail.com>
This commit is contained in:
leejetandnoctrex authored and GitHub committed 2026-11-06 00:35:58 +08:00
1 parent 58bb3c1b1c
commit 7952693004
48 files changed
+2003 -102

No files matched your search

+1 -1
View File
@@ -75,7 +75,7 @@ namespace H3VUpsampler {
const TensorStorage* embed_mid = find_weight("embed.2.weight");
if (embed_mid != nullptr && embed_mid->n_dims >= 1) {
config.embed_dim = static_cast<int>(embed_mid->ne[0]);
config.embed_dim = static_cast<int>(embed_mid->logical_ne0());
}
std::set<int> in_res_idx, in_temporal_idx, out_res_idx, out_temporal_idx;