/nataili

Primary LanguagePythonGNU Affero General Public License v3.0AGPL-3.0

Nataili: Multimodal AI Python Library

PyPI version Downloads

GitHub license

Nataili is a Python library that provides tools for building multimodal AI applications. With its modular design, Nataili makes it easy to use only the tools you need to build custom AI solutions.

Some of the technologies included in Nataili are:

  • AITemplate: Fast diffusion
  • BLIP: Image captioning
  • CLIP: Interrogation (ranking, similarity), Embedding (text, image) with cache support
  • CLIP+MLP Predictor
  • CodeFormer: Post-processing and super-resolution for faces
  • ESRGAN: Image super-resolution
  • GFPGAN: Post-processing for faces
  • LAION-CoCa: Image captioning
  • Stable Diffusion: Image generation
  • SD Upscaler: latent diffusion upscaler for Stable Diffusion's autoencoder
  • Train: Training tools for Stable Diffusion and CLIP+MLP Predictor

Projects using Nataili

Installation

pip install nataili

From source:

git clone https://github.com/Sygil-Dev/nataili
cd nataili
pip install -e .

Documentation