push
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user