Keep compatibility with some custom nodes.

This commit is contained in:
comfyanonymous 2024-05-26 15:37:24 -04:00
parent 9a151b7def
commit 16a493a190
1 changed files with 5 additions and 0 deletions

View File

@ -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()