fix: prevent torch checkpoint offset overflow (#1832)

This commit is contained in:
leejet authored and GitHub committed 2026-07-29 23:16:29 +08:00
1 parent 9cfe2af8f9
commit e92e86fb11
6 files changed
+121 -27

No files matched your search

+3 -3
View File
@@ -131,9 +131,9 @@ public:
ResidencyMode residency_mode,
ggml_backend_t compute_backend,
ggml_backend_t params_backend,
size_t* registered_tensor_size = nullptr,
bool allow_split_buffer = false,
bool params_follow_compute_backend = false,
size_t* registered_tensor_size = nullptr,
bool allow_split_buffer = false,
bool params_follow_compute_backend = false,
const std::map<ggml_tensor*, enum ggml_op>* tensor_ops = nullptr);
bool unregister_param_tensors(const std::string& desc,