hunyuanvideo dit

2024-12-17 14:45:23 +08:00
parent 7c0520d029
commit 05e2028c5d
5 changed files with 706 additions and 6 deletions
--- a/diffsynth/models/utils.py
+++ b/diffsynth/models/utils.py
@@ -80,7 +80,7 @@ def load_state_dict_from_safetensors(file_path, torch_dtype=None):


 def load_state_dict_from_bin(file_path, torch_dtype=None):
-    state_dict = torch.load(file_path, map_location="cpu")
+    state_dict = torch.load(file_path, map_location="cpu", weights_only=True)
    if torch_dtype is not None:
        for i in state_dict:
            if isinstance(state_dict[i], torch.Tensor):