ComfyUI/comfy_extras/chainner_models/model_loading.py
Chenlei Hu d7969cb070
Replace print with logging (#6138)
* Replace print with logging

* nit

* nit

* nit

* nit

* nit

* nit
2024-12-20 16:24:55 -05:00

7 lines
260 B
Python

import logging
from spandrel import ModelLoader
def load_state_dict(state_dict):
logging.warning("comfy_extras.chainner_models is deprecated and has been replaced by the spandrel library.")
return ModelLoader().load_from_state_dict(state_dict).eval()