ComfyUI Extension: ComfyUI-LLM-VLM-Node

Authored by squirrel765

Created 21 days ago

Updated 21 days ago

2 stars

A unified ComfyUI node for both LLM (text-only) and VLM (image-to-text) generation using GGUF models..

Custom Nodes (0)

README

ComfyUI 통합 LLM/VLM 노드

이 저장소는 텍스트 기반 LLM(거대 언어 모델)과 멀티모달 VLM(비전 언어 모델)을 하나의 노드로 통합한 ComfyUI 커스텀 노드를 포함합니다. 이 노드를 사용하면 ComfyUI 워크플로우 내에서 직접 텍스트 생성 및 이미지 설명(이미지-to-텍스트) 작업을 수행할 수 있습니다. 이 프로젝트는 LevelPixel/ComfyUI-LevelPixel-Advanced의 코드 구조에서 영감을 받아 제작되었습니다. 원본 개발자에게 감사를 표합니다.

✨ 주요 기능

통합 노드: 텍스트 생성과 이미지 기반 생성을 위한 단일 노드를 제공합니다.
듀얼 모드: 'Text-Only'와 'Multimodal' 두 가지 모드 간 전환이 가능합니다.
GGUF 모델 지원: llama-cpp-python을 기반으로 하여 인기 있는 GGUF 모델 형식을 사용합니다.
VLM 지원: Llava 1.6 및 Qwen2.5-VL과 같은 비전 모델과 호환되어 이미지를 이해하고 설명할 수 있습니다.
메모리 관리: 생성 후 VRAM 및 RAM을 확보할 수 있는 'Unload' 옵션이 포함되어 있습니다.
세부 설정: Temperature, Top_p, Top_k, 페널티 등 다양한 생성 파라미터를 정밀하게 제어할 수 있습니다.

📦 설치 방법

cd ComfyUI/custom_nodes/
git clone https://github.com/squirrel765/ComfyUI-LLM-VLM-Node.git

ComfyUI에 내장된 Python 환경에 llama-cpp-python을 설치해야 합니다.

🚀 사용 방법

모델 파일 준비

LLM/VLM 모델: .gguf 형식의 주 모델 파일을 ComfyUI/models/LLavacheckpoints/ 폴더에 넣으세요. 폴더가 없다면 새로 생성해야 합니다.

노드 사용법

ComfyUI 메뉴에서 Add Node > LLM/Generators > Unified LLM/VLM Generator를 추가합니다.

모드 (Text-Only , Multimodal):

Text-Only: 텍스트 프롬프트만 사용하여 텍스트를 생성합니다.

Multimodal: 이미지와 텍스트 프롬프트를 함께 사용하여 텍스트를 생성합니다.

ckpt_name: LLavacheckpoints 폴더에 있는 주 GGUF 모델을 선택합니다. image (멀티모달 전용): 다른 노드(예: Load Image)의 이미지 출력을 연결합니다. clip_name (멀티모달 전용): ckpt_name에 맞는 비전(CLIP) 모델을 선택합니다. prompt: 모델에 전달할 텍스트 프롬프트입니다. system_msg: 모델의 역할이나 행동을 정의하는 시스템 프롬프트입니다. gpu_layers: VRAM에 오프로드할 레이어 수를 지정합니다. 사용자의 VRAM 크기에 맞게 조절하세요. (-1은 모든 레이어를 오프로드합니다.) unload: True로 설정하면 생성 작업이 끝난 후 모델을 메모리에서 해제하여 시스템 리소스를 절약합니다.

예시