mirror of
https://github.com/leejet/stable-diffusion.cpp.git
synced 2026-11-10 19:06:33 +00:00
fix: load INT8 convrot LLM embeddings correctly (#2067)
This commit is contained in:
1 parent
47e83d7136
commit
ede32a621e
4 files changed
+91
-12
No files matched your search
@@ -210,7 +210,7 @@ WeightPrefetchResult ModelManager::prefetch_params(
|
||||
is_optional_missing_tensor(state->name)) {
|
||||
continue;
|
||||
}
|
||||
if (state->compute_backend == state->params_backend) {
|
||||
if (state->usage_op == GGML_OP_CUSTOM || state->compute_backend == state->params_backend) {
|
||||
needs_synchronous_load = needs_synchronous_load ||
|
||||
!state->loaded_to_params_backend;
|
||||
continue;
|
||||
@@ -275,6 +275,7 @@ bool ModelManager::activate_prefetched_params(
|
||||
[&](TensorState* state) {
|
||||
return state == nullptr || should_ignore(*state) ||
|
||||
is_optional_missing_tensor(state->name) ||
|
||||
state->usage_op == GGML_OP_CUSTOM ||
|
||||
state->compute_backend == state->params_backend ||
|
||||
state->staged_to_compute_backend;
|
||||
});
|
||||
|
||||
Reference in new issue
Block a user