Aller au contenu
Node packVérifiéExterne
Ouvrir la source

ComfyUI-FL-PixelGen

FL PixelGen - Pixel-space diffusion text-to-image generation and LoRA training nodes for ComfyUI

Mainteneur@filliptmUnratedClasse

Profil du package

Statut
Vérifié
Éditeur
@filliptm
Source
GitHub
Étoiles
13

LIREMOI

FL PixelGen

Pixel-space diffusion text-to-image generation and LoRA training nodes for ComfyUI, powered by PixelGen. Generate high-quality images directly in pixel space using a JiT transformer denoiser with Qwen3 text conditioning and flow matching sampling — entirely within ComfyUI's node graph.

PixelGen ComfyUI Patreon

PixelGen Preview

Features

  • Pixel-Space Diffusion — Generates images directly in pixel space, no VAE decoding needed
  • Live Sampling Preview — See the image form step-by-step during generation
  • REPA LoRA Training — 4-component loss (flow matching + cosine alignment + LPIPS + DINO) with noise gating
  • Real-Time Training UI — Live loss chart, progress bar, validation samples, and lightbox preview via WebSocket widget
  • Qwen3 Text Encoder — Uses Qwen3-1.7B for text conditioning, auto-downloaded from HuggingFace
  • Auto Model Download — Denoiser and text encoder models download automatically from HuggingFace on first use

Nodes

NodeCategoryDescription
FL PixelGen Model LoaderLoadersLoad the PixelGen denoiser model from checkpoint
FL PixelGen Text Encoder LoaderLoadersLoad a Qwen3 text encoder for conditioning
FL PixelGen GenerateGenerationGenerate images from text using pixel-space diffusion
FL PixelGen LoRA LoaderLoadersLoad and apply a trained LoRA adapter for inference
FL PixelGen Training ConfigTrainingConfigure LoRA training hyperparameters and REPA loss weights
FL PixelGen TrainTrainingRun REPA LoRA training with real-time progress widget

Installation

Manual

cd ComfyUI/custom_nodes
git clone https://github.com/filliptm/ComfyUI-FL-PixelGen.git
cd ComfyUI-FL-PixelGen
pip install -r requirements.txt

The training widget JS is included in js/ and requires no build step.

Quick Start

Inference Pipeline

  1. Load Model — Add FL PixelGen Model Loader and select PixelGen-XXL-T2I
  2. Load Text Encoder — Add FL PixelGen Text Encoder Loader and select a Qwen3 variant
  3. Generate — Connect both to FL PixelGen Generate, enter a prompt, and execute

Training Pipeline

  1. Load Model + Text Encoder — Same loader nodes as inference
  2. Configure — Set LoRA rank, learning rate, epochs, and REPA loss weights in FL PixelGen Training Config
  3. Train — Connect model, text_encoder, and config to FL PixelGen Train with your image directory
  4. Validate — Set a validation prompt to see sample outputs during training

Using Trained LoRAs

Connect a trained LoRA checkpoint to FL PixelGen LoRA Loader between the model loader and generate nodes. Adjust strength to control the effect.

Node Details

FL PixelGen Model Loader

Loads the PixelGen denoiser from a local checkpoint or HuggingFace.

InputTypeDefaultNotes
model_nameDropdownPixelGen-XXL-T2IAuto-downloads from HF
deviceDropdowncudacuda / cpu
dtypeDropdownbfloat16bfloat16 / float32
force_reloadBOOLEANFalseBypass model cache

Output: PIXELGEN_MODEL

FL PixelGen Text Encoder Loader

Loads the Qwen3-1.7B language model as the text encoder for conditioning.

InputTypeDefaultNotes
model_nameDropdownQwen/Qwen3-1.7BAuto-downloads from HF

Nœuds dans ce pack

Nœuds 6
  • FL_PixelGen_Generate
  • FL_PixelGen_LoraLoader
  • FL_PixelGen_ModelLoader
  • FL_PixelGen_TextEncoderLoader
  • FL_PixelGen_Train
  • FL_PixelGen_TrainingConfig

Sources

3 sources
GitHubSourceVersion b1e947243c27cf6994d586d3fcd7bdd2fb19acfa
github.com/filliptm/ComfyUI-FL-PixelGen

Vérifié 11 oct.

GitHubSource
github.com/filliptm/ComfyUI-FL-PixelGen

Vérifié 2 nov.

Extraits de sources

1 extrait

Source context: Repo filliptm/ComfyUI-FL-PixelGen