mirror of
https://github.com/leejet/stable-diffusion.cpp.git
synced 2026-11-12 03:49:09 +00:00
feat: support packed INT4 convrot models (#2276)
Co-authored-by: noctrex <noctrex@gmail.com>
This commit is contained in:
1 parent
58bb3c1b1c
commit
7952693004
48 files changed
+2003
-102
No files matched your search
@@ -24,14 +24,14 @@ namespace Qwen {
|
||||
return it == weights.end() ? nullptr : &it->second;
|
||||
};
|
||||
if (auto w = find("img_in.weight")) {
|
||||
config.in_channels = w->ne[0];
|
||||
config.in_channels = w->logical_ne0();
|
||||
config.hidden_size = w->ne[1];
|
||||
}
|
||||
if (auto w = find("proj_out.weight")) {
|
||||
config.out_channels = w->ne[1];
|
||||
}
|
||||
if (auto w = find("txt_in.in_layer.weight")) {
|
||||
config.context_dim = w->ne[0];
|
||||
config.context_dim = w->logical_ne0();
|
||||
}
|
||||
if (auto w = find("transformer_blocks.0.attn.norm_q.weight")) {
|
||||
config.head_dim = w->ne[0];
|
||||
|
||||
Reference in new issue
Block a user