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

fix: break the line after skipping tensors in VAE (#591)

Browse Source
This commit is contained in:
vmobilis 2025-07-03 17:50:42 +03:00 committed by GitHub
parent 8d0819c548
commit 3bae667f3d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
model.cpp
View File

@ -1841,6 +1841,7 @@ bool ModelLoader::load_tensors(on_new_tensor_cb_t on_new_tensor_cb, ggml_backend
}; };
int tensor_count = 0; int tensor_count = 0;
int64_t t1 = ggml_time_ms(); int64_t t1 = ggml_time_ms();
bool partial = false;
for (auto& tensor_storage : processed_tensor_storages) { for (auto& tensor_storage : processed_tensor_storages) {
if (tensor_storage.file_index != file_index) { if (tensor_storage.file_index != file_index) {
++tensor_count; ++tensor_count;
@ -1922,15 +1923,21 @@ bool ModelLoader::load_tensors(on_new_tensor_cb_t on_new_tensor_cb, ggml_backend
ggml_backend_tensor_set(dst_tensor, convert_buffer.data(), 0, ggml_nbytes(dst_tensor)); ggml_backend_tensor_set(dst_tensor, convert_buffer.data(), 0, ggml_nbytes(dst_tensor));
} }
} }
size_t tensor_max = processed_tensor_storages.size();
int64_t t2 = ggml_time_ms(); int64_t t2 = ggml_time_ms();
pretty_progress(++tensor_count, processed_tensor_storages.size(), (t2 - t1) / 1000.0f); pretty_progress(++tensor_count, tensor_max, (t2 - t1) / 1000.0f);
t1 = t2; t1 = t2;
partial = tensor_count != tensor_max;
} }
if (zip != NULL) { if (zip != NULL) {
zip_close(zip); zip_close(zip);
} }
if (partial) {
printf("\n");
}
if (!success) { if (!success) {
break; break;
} }