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

Support LoRA encoder (#695)

Browse Source 
* lora_encoder
This commit is contained in:
Zhongjie Duan
2025-07-19 20:44:03 +08:00
committed by GitHub
parent d19fcc8c04
commit 1384de0353
7 changed files with 294 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
diffsynth/models/utils.py
View File

@@ -71,7 +71,7 @@ def load_state_dict(file_path, torch_dtype=None, device="cpu"):
def load_state_dict_from_safetensors(file_path, torch_dtype=None, device="cpu"):
state_dict = {}
with safe_open(file_path, framework="pt", device=device) as f:
with safe_open(file_path, framework="pt", device=str(device)) as f:
for k in f.keys():
state_dict[k] = f.get_tensor(k)
if torch_dtype is not None: