refactor: move model file IO into dedicated module (#1442)

This commit is contained in:
leejet authored and GitHub committed 2026-04-19 17:52:56 +08:00
1 parent 7023fc4cfb
commit 66143340b6
13 files changed
+1022 -818

No files matched your search

+14
View File
@@ -0,0 +1,14 @@
#ifndef __SD_MODEL_IO_SAFETENSORS_IO_H__
#define __SD_MODEL_IO_SAFETENSORS_IO_H__
#include <string>
#include <vector>
#include "tensor_storage.h"
bool is_safetensors_file(const std::string& file_path);
bool read_safetensors_file(const std::string& file_path,
std::vector<TensorStorage>& tensor_storages,
std::string* error = nullptr);
#endif // __SD_MODEL_IO_SAFETENSORS_IO_H__