ComfyUI-PixArt_XL

ComfyUI-PixArt_XL
★ 3

模型集成内存管理分辨率回退调度器支持
将PixArt-Alpha XL模型无缝集成到ComfyUI，提供GPU与系统内存智能管理、自动分辨率回退与高质量放大，并支持多种调度器与全精度稳定处理。
💡 在ComfyUI中用PixArt-Alpha XL生成大分辨率图像并自动降级以节省显存。
🍴 1 Forks💻 Python🔄 2025-03-20
📦
网盘下载
复制链接后前往夸克网盘下载
https://pan.quark.cn/s/8f9eee5e2cdb
📦 requirements.txt
diffusers
transformers
sentencepiece
Buy Me A Coffee
image
📄 README

ComfyUI-PixArt_XL

Support My Work

If you find this project helpful, consider buying me a coffee:

[](https://buymeacoffee.com/shmuelronen)

A ComfyUI extension that integrates PixArt-Alpha models directly into ComfyUI with advanced memory management.

Features

  • Use PixArt-Alpha XL models directly in ComfyUI
  • Smart memory management using both GPU and system RAM
  • Automatic fallback to lower resolutions with high-quality upscaling
  • Support for both DPM-Solver and SA-Solver schedulers
  • Full precision processing for maximum stability

    • Installation

  • Clone this repository into your ComfyUI custom nodes folder:
    • cd ComfyUI/custom_nodes
git clone https://github.com/ShmuelRonen/ComfyUI-PixArt_XL.git

  • Install dependencies:
    • pip install sentencepiece

  • Restart ComfyUI

    • Usage

    After installation, you’ll have access to two new nodes:

    PixArtAlpha ModelLoader

    Automatic loads a PixArt-Alpha model from Hugging Face.

    Inputs:

  • base_model_path: Path to the model (default: “PixArt-alpha/PixArt-XL-2-1024-MS”)

    • Outputs:

  • model: The loaded model

    • PixArtAlpha Generation

    Generates images using the PixArt-Alpha model.

    Inputs:

  • model: The PixArt-Alpha model
  • positive: Positive prompt
  • negative: Negative prompt
  • width, height: Image dimensions
  • steps: Number of sampling steps
  • guidance_scale: How closely to follow the prompt
  • schedule: Sampling scheduler
  • seed: Random seed

    • Outputs:

  • image: The generated image

    • Memory Optimization

    This implementation includes several advanced techniques to maximize memory efficiency:

  • CPU component offloading
  • Attention slicing
  • VAE optimization
  • Periodic CUDA cache clearing
  • Resolution fallback with quality-preserving upscaling

    • Acknowledgments

  • PixArt-alpha
  • Diffusers library
  • ComfyUI