Skip to content

Commit

Permalink
chore: make code cleaner
Browse files Browse the repository at this point in the history
  • Loading branch information
leejet committed Dec 9, 2023
1 parent 2eac844 commit 69efe3c
Show file tree
Hide file tree
Showing 3 changed files with 145 additions and 248 deletions.
3 changes: 2 additions & 1 deletion model.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -1102,6 +1102,7 @@ bool ModelLoader::parse_data_pkl(uint8_t* buffer,
reader.tensor_storage.file_index = file_index;
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());
// reset
reader = PickleTensorReader();
}
Expand Down Expand Up @@ -1139,7 +1140,7 @@ bool ModelLoader::init_from_ckpt_file(const std::string& file_path, const std::s
size_t pkl_size;
zip_entry_read(zip, &pkl_data, &pkl_size);

LOG_DEBUG("%lld", pkl_size);
// LOG_DEBUG("%lld", pkl_size);

parse_data_pkl((uint8_t*)pkl_data, pkl_size, zip, dir, file_index, prefix);

Expand Down
2 changes: 1 addition & 1 deletion model.h
Original file line number Diff line number Diff line change
Expand Up @@ -7,8 +7,8 @@
#include <string>
#include <vector>

#include "ggml/ggml.h"
#include "ggml/ggml-backend.h"
#include "ggml/ggml.h"
#include "json.hpp"
#include "zip.h"

Expand Down
Loading

0 comments on commit 69efe3c

Please sign in to comment.