VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-08 22:35:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'kohya-ss:main' into feature/stratified_lr

Browse Source
This commit is contained in:
u-haru
2023-03-31 12:47:56 +09:00
committed by GitHub
parent 94441fa746 c93cbbc373
commit 41ecccb2a9
3 changed files with 21 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
library/model_util.py
View File

@@ -841,7 +841,7 @@ def load_checkpoint_with_text_encoder_conversion(ckpt_path, device="cpu"):
if is_safetensors(ckpt_path):
checkpoint = None
state_dict = load_file(ckpt_path, device)
state_dict = load_file(ckpt_path) # , device) # may causes error
else:
checkpoint = torch.load(ckpt_path, map_location=device)
if "state_dict" in checkpoint: