upstream-archive/stable-diffusion.cpp
1
0
Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2026-07-29 14:20:57 -05:00
Code Issues Packages Projects Releases Wiki Activity

first efforts at implementing photomaker; lots more to do

Browse Source
This commit is contained in:
bssrdf
2024-02-02 18:36:40 -05:00
parent 349439f239
commit 5dbbea0ca4
7 changed files with 369 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
model.cpp
View File

@@ -788,7 +788,6 @@ bool ModelLoader::init_from_safetensors_file(const std::string& file_path, const
for (int i = 0; i < n_dims; i++) {
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);
tensor_storage.reverse_ne();
@@ -1127,6 +1126,8 @@ bool ModelLoader::parse_data_pkl(uint8_t* buffer,
if (reader.phase == PickleTensorReader::READ_DIMENS) {
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());
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());