reworked pmid lora

This commit is contained in:
bssrdf
2024-02-12 18:34:13 -05:00
parent b0579ec460
commit 753231a7b4
3 changed files with 39 additions and 230 deletions

View File

@@ -790,7 +790,7 @@ bool ModelLoader::init_from_safetensors_file(const std::string& file_path, const
}
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));
// printf(" %s:%s [%zu, %zu, %zu, %zu]\n", (prefix + name).c_str(), ggml_type_name(type), ne[0], ne[1],ne[2], ne[3]);
tensor_storage.reverse_ne();
size_t tensor_data_size = end - begin;