test/comfy_extras/chainner_models/model_loading.py

6 lines
244 B
Python
Raw Normal View History

2024-08-03 08:53:56 +00:00
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()