DarthAffe/stable-diffusion.cpp
1
0
Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2025-12-13 05:48:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

add wan gguf support

Browse Source
This commit is contained in:
leejet 2025-08-16 00:46:49 +08:00
parent 3a2840f9fb
commit b0833eb4d8
1 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
model.cpp
View File

@ -1038,7 +1038,22 @@ bool ModelLoader::init_from_gguf_file(const std::string& file_path, const std::s
gguf_context* ctx_gguf_ = NULL;
ggml_context* ctx_meta_ = NULL;
ctx_gguf_ = gguf_init_from_file(file_path.c_str(), {true, &ctx_meta_});
auto on_tensor_shape_read = [](const int64_t* ne, uint32_t n_dims, struct gguf_tensor_shape* shape) -> bool {
for (int i = 0; i < GGML_MAX_DIMS; i++) {
if (i < n_dims) {
shape->ne[i] = ne[i];
} else {
shape->ne[i] = 1;
}
}
for (int i = GGML_MAX_DIMS; i < n_dims; i++) {
shape->ne[GGML_MAX_DIMS - 1] *= ne[i]; // stack to last dim;
}
return true;
};
ctx_gguf_ = gguf_init_from_file_ext(file_path.c_str(), {true, &ctx_meta_}, on_tensor_shape_read);
if (!ctx_gguf_) {
LOG_ERROR("failed to open '%s'", file_path.c_str());
return false;