From 537f35c7bc76e611b53102d3b5a2926e46ec7362 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Tue, 2 Jul 2024 20:21:51 -0400
Subject: [PATCH] Don't update dict if contiguous.

---
 comfy/sd.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/comfy/sd.py b/comfy/sd.py
index b28bfe358..4ba00a62c 100644
--- a/comfy/sd.py
+++ b/comfy/sd.py
@@ -627,7 +627,10 @@ def save_checkpoint(output_path, model, clip=None, vae=None, clip_vision=None, m
     sd = model.model.state_dict_for_saving(clip_sd, vae.get_sd(), clip_vision_sd)
     for k in extra_keys:
         sd[k] = extra_keys[k]
+
     for k in sd:
-        sd[k] = sd[k].contiguous()
+        t = sd[k]
+        if not t.is_contiguous():
+            sd[k] = t.contiguous()
 
     comfy.utils.save_torch_file(sd, output_path, metadata=metadata)