theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-21 16:18:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

hunyuanvideo dit

Browse Source
This commit is contained in:
Artiprocher
2024-12-17 14:45:23 +08:00
parent 7c0520d029
commit 05e2028c5d
5 changed files with 706 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
diffsynth/models/utils.py
View File

@@ -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):