DarthAffe/stable-diffusion.cpp
1
0 Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2026-11-11 03:24:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

more progress; now alloc buffer failed

Browse Source
This commit is contained in:
bssrdf committed 2024-02-10 22:53:54 -05:00
1 parent d93951456d
commit 807c340118
5 files changed
+55 -15
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

model.cpp
+3 -1
View File
@@ -789,6 +789,8 @@ bool ModelLoader::init_from_safetensors_file(const std::string& file_path, const
ne[i] = shape[i].get<int64_t>();
}
TensorStorage tensor_storage(prefix + name, type, ne, n_dims, file_index, ST_HEADER_SIZE_LEN + header_size_ + begin);
// if(strcmp(prefix.c_str(), "pmid") == 0)
// printf(" %s %s \n", (prefix + name).c_str(), ggml_type_name(type));
tensor_storage.reverse_ne();
size_t tensor_data_size = end - begin;
@@ -1126,7 +1128,7 @@ bool ModelLoader::parse_data_pkl(uint8_t* buffer,
reader.tensor_storage.reverse_ne();
reader.tensor_storage.file_index = file_index;
// if(strcmp(prefix.c_str(), "pm") == 0)
printf(" got tensor %s \n ", reader.tensor_storage.name.c_str());
// printf(" got tensor %s \n ", reader.tensor_storage.name.c_str());
reader.tensor_storage.name = prefix + reader.tensor_storage.name;
tensor_storages.push_back(reader.tensor_storage);
// LOG_DEBUG("%s", reader.tensor_storage.name.c_str());