From 16a493a19042227baadd939fc095305716ae58db Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Sun, 26 May 2024 15:37:24 -0400 Subject: [PATCH] Keep compatibility with some custom nodes. --- comfy_extras/chainner_models/model_loading.py | 5 +++++ 1 file changed, 5 insertions(+) create mode 100644 comfy_extras/chainner_models/model_loading.py diff --git a/comfy_extras/chainner_models/model_loading.py b/comfy_extras/chainner_models/model_loading.py new file mode 100644 index 00000000..d48bc238 --- /dev/null +++ b/comfy_extras/chainner_models/model_loading.py @@ -0,0 +1,5 @@ +from spandrel import ModelLoader + +def load_state_dict(state_dict): + print("WARNING: comfy_extras.chainner_models is deprecated and has been replaced by the spandrel library.") + return ModelLoader().load_from_state_dict(state_dict).eval()