1.6 KiB
ExLlama nodes for ComfyUI
A simple prompt generator for ComfyUI utilizing ExLlama.
Outputs are printed in the console, sadly I have no idea how to append them to metadata or display in the UI.
Suggestions welcome.
Installation
Clone the repository to custom_nodes in your ComfyUI directory:
git clone https://github.com/Zuellni/ComfyUI-ExLlama-Nodes
Install the latest ExLlama package from https://github.com/jllllll/exllama/releases.
Choose the version matching your platform, Python, and PyTorch CUDA/ROCm.
Example for Windows with Python 3.10 and CUDA 11.7:
pip install https://github.com/jllllll/exllama/releases/download/0.0.17/exllama-0.0.17+cu117-cp310-cp310-win_amd64.whl
Nodes
Comes with the following nodes:
ExLlama Loader
Used to load 4-bit GPTQ Llama/2 models. You can find a lot of them over at Hugging Face.
ExLlama allocates memory according to max_seq_len. Lowering it is a good way to save on GPU RAM.
It's not possible to offload the model to CPU RAM currently.
ExLlama Generator
Generates a string based on the given prompt for use with other nodes.
Default parameter values correspond to the simple-1 preset from text-generation-webui.
ExLlama isn't deterministic, so the outputs may differ even with the same seed.
Example
The workflow Can be loaded directly in ComfyUI.