diff --git a/core/generator.py b/core/generator.py index 36a5510..8298dc3 100644 --- a/core/generator.py +++ b/core/generator.py @@ -16,6 +16,16 @@ from comfy.utils import ( logger = logging.getLogger(__name__) + +class _SuppressModelConfigWarning(logging.Filter): + """Suppress the expected 'Could not read model config' error from SGLang when loading safetensors files.""" + def filter(self, record): + return "Could not read model config" not in record.getMessage() + + +logging.getLogger("sglang.multimodal_gen.registry").addFilter(_SuppressModelConfigWarning()) + + try: from sglang.multimodal_gen import DiffGenerator except ImportError: