refine sd.py

This commit is contained in:
Yao Chi 2024-08-25 23:49:45 +08:00
parent 949df83234
commit 9d2fcc7d06

View File

@ -696,4 +696,4 @@ def save_checkpoint(output_path, model, clip=None, vae=None, clip_vision=None, m
if not t.is_contiguous(): if not t.is_contiguous():
sd[k] = t.contiguous() sd[k] = t.contiguous()
comfy.utils.save_torch_file(sd, output_path, metadata=metadata) comfy.utils.save_torch_file(sd, output_path, metadata=metadata)