From 2eefd1c1b316a6392a3a549cc35e8ffe4d46be2f Mon Sep 17 00:00:00 2001
From: obisin <90648621+obisin@users.noreply.github.com>
Date: Thu, 4 Sep 2025 16:00:07 +0100
Subject: [PATCH] Update dgls_model_loader.py

general bug fixes
---
 dgls_model_loader.py | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/dgls_model_loader.py b/dgls_model_loader.py
index 65edcf7..e5a92c0 100644
--- a/dgls_model_loader.py
+++ b/dgls_model_loader.py
@@ -180,7 +180,7 @@ class DGLSModelLoader:
                 "verbose": ("BOOLEAN", {"default": False}),
             },
             "optional": {"nuke_all_caches": ("BOOLEAN", {"default": False, "tooltip": "CAUTION: Aggressive Clear all ComfyUI caches"}),
-                "custom_ckpt_path": ("STRING", {"default": ""}),
+                # "custom_ckpt_path": ("STRING", {"default": ""}),
             }
         }
 
@@ -190,11 +190,9 @@ class DGLSModelLoader:
     CATEGORY = "loaders"
     TITLE = "DGLS Model Loader"
 
-    def load_dgls_model(self, model_name, model_type, cast_dtype,  verbose, clear_model_cache=False, nuke_all_caches=False, custom_ckpt_path="", override_model_type="auto"):
-
+    def load_dgls_model(self, model_name, model_type, cast_dtype,  verbose, clear_model_cache, nuke_all_caches=False, custom_ckpt_path="", override_model_type="auto"):
 
         self.verbose = verbose
-
         # Cache clearing logic
         if nuke_all_caches:
             self.nuke_all_caches()
@@ -297,8 +295,7 @@ class DGLSModelLoader:
                         comfy.utils.set_attr_param(
                             model_patcher.model,
                             name,
-                            param.data.clone()
-                        )
+                            param.data.clone())
         return model_patcher
 
     def normalize_inference_model_patcher(self, model_patcher, verbose=False, attr_names=None):