fix: try disabling grad on chkpoint load

Haidra-Org · Mar 12, 2024 · f3d5cc0 · f3d5cc0
1 parent d13e782
commit f3d5cc0
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/hordelib/nodes/node_model_loader.py b/hordelib/nodes/node_model_loader.py
@@ -94,12 +94,13 @@ def load_checkpoint(
         SharedModelManager.manager._models_in_ram = {}
 
         ckpt_path = folder_paths.get_full_path("checkpoints", ckpt_name)
-        result = comfy.sd.load_checkpoint_guess_config(
-            ckpt_path,
-            output_vae=True,
-            output_clip=True,
-            embedding_directory=folder_paths.get_folder_paths("embeddings"),
-        )
+        with torch.no_grad():
+            result = comfy.sd.load_checkpoint_guess_config(
+                ckpt_path,
+                output_vae=True,
+                output_clip=True,
+                embedding_directory=folder_paths.get_folder_paths("embeddings"),
+            )
 
         SharedModelManager.manager._models_in_ram[horde_in_memory_name] = result, will_load_loras