ComfyUiRagCustomNodes

This package contains custom nodes for ComfyUI: Image generation using Stable Diffusion and LLM, Uploading and searching images via ChromaDB and MinIO (S3), Integration…

By Santat2023·Created 11 months ago·Updated 10 months ago· 0

Santat2023/ComfyUiRagNodes

Nodes—

On cloudLocal install

Stars0

Updated10 months ago

Readme

ComfyUiRagCustomNodes

Этот пакет содержит кастомные ноды для ComfyUI:

Генерация изображений с помощью Stable Diffusion и LLM
Загрузка и поиск изображений через Qdrant и MinIO (S3)
Интеграция с CLIP, Ollama, и другими инструментами

Структура

custom_ksampler.py — кастомный KSampler с поддержкой RAG
llm_node.py — нода для работы с LLM (Ollama)
db_load_node.py — загрузка изображений из базы
s3_utils.py — утилиты для работы с MinIO/S3
load_images_utils.py — утилиты для поиска и загрузки изображений

Установка

Скопируйте папку в директорию custom_nodes вашего ComfyUI.

Требования

Python 3.10+
ComfyUI
MinIO/S3
Qdrant
Ollama

Поддержка Ollama

Для использования LLM ноды необходимо установить Ollama.

Установка Ollama

Windows/MacOS/Linux: скачайте установщик с официального сайта

Установка LLM

LLM нода работает с моделью qwen2:7b, для ее установки:

ollama pull qwen2:7b

Использование

Добавьте ноды через интерфейс ComfyUI и настройте параметры согласно вашим задачам.

Примеры

В папке testFlows содержатся примеры применения кастомных pipelines. Для их корректной работы необходимо установить дополнительный репозиторий: ComfyUI_IPAdapter_plus