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

fix crashing in saving lora with clipskip

Browse Source
This commit is contained in:
Kohya S
2023-08-08 21:03:16 +09:00
parent 7ea38f90d7
commit 6f80fe17fc
2 changed files with 18 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
library/sai_model_spec.py
View File

@@ -215,7 +215,11 @@ def build_metadata(
metadata["modelspec.prediction_type"] = PRED_TYPE_EPSILON metadata["modelspec.prediction_type"] = PRED_TYPE_EPSILON
if timesteps is not None: if timesteps is not None:
metadata["modelspec.timestep_range"] = timesteps if isinstance(timesteps, str) or isinstance(timesteps, int):
timesteps = (timesteps, timesteps)
if len(timesteps) == 1:
timesteps = (timesteps[0], timesteps[0])
metadata["modelspec.timestep_range"] = f"{timesteps[0]},{timesteps[1]}"
else: else:
del metadata["modelspec.timestep_range"] del metadata["modelspec.timestep_range"]

18
library/train_util.py
View File

@@ -2546,15 +2546,15 @@ def get_sai_model_spec(
lora, lora,
textual_inversion, textual_inversion,
timestamp, timestamp,
title, title=title,
reso, reso=reso,
is_stable_diffusion_ckpt, is_stable_diffusion_ckpt=is_stable_diffusion_ckpt,
args.metadata_author, author=args.metadata_author,
args.metadata_description, description=args.metadata_description,
args.metadata_license, license=args.metadata_license,
args.metadata_tags, tags=args.metadata_tags,
timesteps, timesteps=timesteps,
args.clip_skip, # None or int clip_skip=args.clip_skip, # None or int
) )
return metadata return metadata