Adds two nodes (FL VLM Model Loader, FL VLM Image Inference) with a Qwen3-VL backend, optional 8-bit/4-bit quantization, resize presets, per-image or shared-prompt batch modes, and a streaming progress UI. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
41 lines
1020 B
Python
41 lines
1020 B
Python
import logging
|
|
import os
|
|
|
|
|
|
logger = logging.getLogger("FL_VLM")
|
|
logger.setLevel(logging.INFO)
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
try:
|
|
import nodes
|
|
|
|
current_dir = os.path.dirname(os.path.abspath(__file__))
|
|
web_dir = os.path.join(current_dir, "web")
|
|
if os.path.exists(web_dir):
|
|
nodes.EXTENSION_WEB_DIRS["ComfyUI-FL-VLM"] = web_dir
|
|
except Exception as e:
|
|
logger.warning("Could not register FL VLM web directory: %s", e)
|
|
|
|
try:
|
|
from .modules.loader import register_vlm_folder
|
|
|
|
register_vlm_folder()
|
|
except Exception as e:
|
|
logger.warning("Could not register FL VLM model folder: %s", e)
|
|
|
|
try:
|
|
from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
|
|
|
|
print("FL VLM Custom Nodes Loaded - Version 0.1.0")
|
|
except Exception as e:
|
|
logger.error("Failed to load FL VLM nodes: %s", e)
|
|
import traceback
|
|
|
|
traceback.print_exc()
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|