feat: support safetensors export in convert mode (#1444)

This commit is contained in:
leejet authored and GitHub committed 2026-04-20 00:22:11 +08:00
1 parent 0a7ae07f94
commit 44cca3d626
10 files changed
+257 -120

No files matched your search

+3
View File
@@ -10,5 +10,8 @@ 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);
bool write_safetensors_file(const std::string& file_path,
const std::vector<TensorWriteInfo>& tensors,
std::string* error = nullptr);
#endif // __SD_MODEL_IO_SAFETENSORS_IO_H__