19 lines
905 B
Python
19 lines
905 B
Python
from .vlm_gguf import NODE_CLASS_MAPPINGS as _vlm_mappings
|
|
from .vlm_gguf import NODE_DISPLAY_NAME_MAPPINGS as _vlm_names
|
|
from .vram_purge import NODE_CLASS_MAPPINGS as _purge_mappings
|
|
from .vram_purge import NODE_DISPLAY_NAME_MAPPINGS as _purge_names
|
|
from .muse_node import NODE_CLASS_MAPPINGS as _muse_mappings
|
|
from .muse_node import NODE_DISPLAY_NAME_MAPPINGS as _muse_names
|
|
|
|
try:
|
|
from .vlm_hf import NODE_CLASS_MAPPINGS as _hf_mappings
|
|
from .vlm_hf import NODE_DISPLAY_NAME_MAPPINGS as _hf_names
|
|
except Exception as _exc:
|
|
print(f"[VLMPrompt] HF backend unavailable (transformers not installed?): {_exc}")
|
|
_hf_mappings, _hf_names = {}, {}
|
|
|
|
NODE_CLASS_MAPPINGS = {**_vlm_mappings, **_purge_mappings, **_muse_mappings, **_hf_mappings}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {**_vlm_names, **_purge_names, **_muse_names, **_hf_names}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|