ComfyUI/comfy_extras/chainner_models/model_loading.py

6 lines
244 B
Python
Raw Permalink Normal View History

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