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

+2 -1
View File
@@ -200,7 +200,8 @@ protected:
bool init_from_gguf_file(const std::string& file_path, const std::string& prefix = "");
bool init_from_safetensors_file(const std::string& file_path, const std::string& prefix = "");
bool init_from_ckpt_file(const std::string& file_path, const std::string& prefix = "");
bool init_from_torch_zip_file(const std::string& file_path, const std::string& prefix = "");
bool init_from_torch_legacy_file(const std::string& file_path, const std::string& prefix = "");
bool init_from_diffusers_file(const std::string& file_path, const std::string& prefix = "");
public: