Skip to content

ComfyUI-IF_AI_tools is a set of custom nodes for ComfyUI that allows you to generate prompts using a local Large Language Model (LLM) via Ollama. This tool enables you to enhance your image generation workflow by leveraging the power of language models.

Notifications You must be signed in to change notification settings

if-ai/ComfyUI-IF_AI_tools

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ComfyUI-IF_AI_tools

ComfyUI-IF_AI_tools

ComfyUI-IF_AI_tools is a set of custom nodes for ComfyUI that allows you to generate prompts using a local Large Language Model (LLM) via Ollama. This tool enables you to enhance your image generation workflow by leveraging the power of language models.

Features

-[NEW] Endpoints for Gemini, LlamaCpp and Mistral

-[NEW] Omost_tool the first tool

Omost_tool

ollama run impactframes/dolphin_llama3_omost Omost via Ollama can be 2x to 3x faster than other Omost servings

Omost Guide

You can watch it here.

Note: In the event of failed outputs it might help to set the temperature below 0.5 as the LLMs can become creative enough to brake the response output format but is normaly safe up to 0.8

You are going to need the comfyui omost nodes here

-https://github.com/huchenlei/ComfyUI_omost?tab=readme-ov-file

-https://github.com/huchenlei/ComfyUI_densediffusion

omost_00004_

LLama3 and Phi3 IF_AI Prompt mkr models released ollama run impactframes/llama3_ifai_sd_prompt_mkr_q4km:latest

ollama run impactframes/ifai_promptmkr_dolphin_phi3:latest

https://huggingface.co/impactframes/llama3_if_ai_sdpromptmkr_q4km

https://huggingface.co/impactframes/ifai_promptmkr_dolphin_phi3_gguf

Prerequisites

  • Ollama - install Ollama. Visit ollama.com for more information.

  • Optionally Kobold.cpp, Oobabooga Llama.cpp or LM Studio. (vision not supported for Oobabooga and Kobold)

  • For optional Apis Set enviromnet variables for "ANTHROPIC_API_KEY", "GEMINI_API_KEY", "OPENAI_API_KEY", "MISTRAL_API _KEY" & "GROQ_API_KEY" with those names or otherwise it won't pick it up and the respective API keys

You can use a .env in the custom_nodes/ComfyUI-IF_AI_tools/.env to define the variables with the same names as above or use the external api_key field on the node

Installation

  1. Install Ollama by following the instructions on their GitHub page on windows

You can also install the Node from the ComfyUI manager

  1. Open a terminal and type following command to install the model:

       ollama run impactframes/llama3_ifai_sd_prompt_mkr_q4km:latest
  2. Move the IF_AI folder from the ComfyUI-IF_AI_tools to inside the root input ComfyUI/input/IF_AI

  3. Navigate to your ComfyUI custom_nodes folder, type CMD on the address bar to open a command prompt, and run the following command to clone the repository:

       git clone https://github.com/if-ai/ComfyUI-IF_AI_tools.git
  4. In ComfyUI protable version just dounle click embedded_install.bat or type CMD on the address bar on the newly created custom_nodes\ComfyUI-IF_AI_tools folder type

       H:\ComfyUI_windows_portable\python_embeded\python.exe -m pip install -r requirements.txt

    replace C:\ for your Drive letter where you have the ComfyUI_windows_portable directory

    On custom environment activate the environment and move to the newly created ComfyUI-IF_AI_tools

       cd ComfyUI-IF_AI_tools
       python -m pip install -r requirements.txt

Usage

  1. Start ComfyUI.

  2. Load the custom workflow located in the custom_nodes\ComfyUI-IF_AI_tools\workflows folder.

  3. Run the queue to generate an image.

Recommended Models

Support

If you find this tool useful, please consider supporting my work by:

Related Tools

  • IF_prompt_MKR -
  • A similar tool available for Stable Diffusion WebUI

AIFuzz made a great video usining ollama and IF_AI tools

AIFuzz

Also Future thinker @ Benji Thankyou both for putting out this awesome videos

Future Thinker @Benji

Example using normal Model

ancient Megastructure, small lone figure 'A dwarfed figure standing atop an ancient megastructure, worn stone towering overhead. Underneath the dim moonlight, intricate engravings adorn the crumbling walls. Overwhelmed by the sheer size and age of the structure, the small figure appears lost amidst the weathered stone behemoth. The background reveals a dark landscape, dotted with faint twinkles from other ancient structures, scattered across the horizon. The silent air is only filled with the soft echoes of distant whispers, carrying secrets of times long past. ethereal-fantasy-concept-art, magical-ambiance, magnificent, celestial, ethereal-lighting, painterly, epic, majestic, dreamy-atmosphere, otherworldly, mystic-elements, surreal, immersive-detail' IF_prompt_Mkr__00011 IF_prompt_Mkr__00012 IF_prompt_Mkr__00010 IF_prompt_Mkr__00014

:IFAItools_comfy

About

ComfyUI-IF_AI_tools is a set of custom nodes for ComfyUI that allows you to generate prompts using a local Large Language Model (LLM) via Ollama. This tool enables you to enhance your image generation workflow by leveraging the power of language models.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published