Matt Miller 94ee49b161 harden: load training-dataset shards with weights_only=True (#14543) ...

LoadTrainingDataset was the only torch.load call in the codebase without
weights_only=True; comfy/utils.py and comfy/sd1_clip.py already pass it.
Recent PyTorch defaults to weights_only=True, so this is defense-in-depth
for installs pinned to older PyTorch. Verified a typical shard (latents +
standard conditioning) round-trips cleanly under weights_only=True.

2026-06-18 15:30:57 -04:00

58 KiB

Raw Permalink Blame History

View Raw