feat: add restricted torch legacy checkpoint loading (#1443)

This commit is contained in:
leejet authored and GitHub committed 2026-04-19 23:09:43 +08:00
1 parent 66143340b6
commit 0a7ae07f94
14 files changed
+1647 -480

No files matched your search

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