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

Fix device issue in load_file, reduce vram usage

Browse Source
This commit is contained in:
Kohya S
2023-03-31 09:05:51 +09:00
parent ea1cf4acee
commit 8cecc676cf
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):
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: