mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-10 06:10:50 +08:00
18 lines
680 B
Python
18 lines
680 B
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
from importlib.resources import files
|
|
from pathlib import Path
|
|
|
|
KNOWN_CHAT_TEMPLATES = {}
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
def _update_known_chat_templates():
|
|
try:
|
|
_chat_templates = files(__package__) / "chat_templates"
|
|
_extra_jinja_templates = {Path(traversable.name).stem: traversable.read_text().replace(' ', '').replace('\n', '') for traversable in _chat_templates.iterdir() if traversable.is_file()}
|
|
KNOWN_CHAT_TEMPLATES.update(_extra_jinja_templates)
|
|
except ImportError as exc:
|
|
logger.warning("Could not load extra chat templates, some text models will fail", exc_info=exc)
|