RomanNum3ral
/
ai-toolkit
mirror of https://github.com/ostris/ai-toolkit.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: exclude LoRA Manager sidecar .metadata.json from resume scan 

get_latest_save_path() uses a broad glob pattern (f"{name}*{post}")
that matches .metadata.json sidecar files written by model managers
like the ComfyUI LoRA Manager. These sidecars have a later ctime
than the matching .safetensors checkpoints, so max() picks them
as the latest save path.

When torch.load() is then called on the .metadata.json file,
it fails with:
  WeightsUnpickler error: Unsupported operand 123
because JSON is not a valid pickle/safetensors format.

Fix: filter out .metadata.json from the candidate paths list,
following the same pattern as the existing false-positive filters.
This commit is contained in:
shane 2026-07-30 00:57:28 +00:00
parent aecd554128
commit 98ffc8446e
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jobs/process/BaseSDTrainProcess.py
View File

@ -823,6 +823,8 @@ class BaseSDTrainProcess(BaseTrainProcess):
paths = [p for p in paths if '_t2i' not in p] paths = [p for p in paths if '_t2i' not in p]
if '_cn' not in name: if '_cn' not in name:
paths = [p for p in paths if '_cn' not in p] paths = [p for p in paths if '_cn' not in p]
# Exclude sidecar metadata written by model managers (e.g. ComfyUI LoRA Manager)
paths = [p for p in paths if not p.endswith('.metadata.json')]
if len(paths) > 0: if len(paths) > 0:
latest_path = max(paths, key=os.path.getctime) latest_path = max(paths, key=os.path.getctime)