Keep compatibility with some custom nodes.

16a493a1 · comfyanonymous · 9a151b7d · 16a493a1
Commit 16a493a1 authored May 26, 2024 by comfyanonymous
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

comfy_extras/chainner_models/model_loading.py comfy_extras/chainner_models/model_loading.py +5 -0

No files found.
--- a/comfy_extras/chainner_models/model_loading.py
+++ b/comfy_extras/chainner_models/model_loading.py
+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()