Tavris1/AI-Toolkit-Easy-Install

One-click Portable Windows installation of 'AI-Toolkit by Ostris'

What it solves

Setting up a diffusion model training environment is notoriously fiddly: you need matching versions of Python, PyTorch, CUDA, FlashAttention, FFmpeg, and more. This project packages those dependencies into a single downloadable ZIP containing one .bat file. Running it installs or updates everything needed and launches the AI-Toolkit GUI/local server without touching your system Python or creating virtual environments.

How it works

You download the release ZIP, extract the batch file into a fresh folder, and double-click it. The installer checks for and installs Git, Node.js, FFmpeg, an embedded Python 3.12.10, PyTorch 2.13.0 with CUDA 13.0, Triton, FlashAttention, torchcodec, the huggingface_hub extension, and finally the latest AI-Toolkit itself. Once ready, the launcher starts the server, opens your default browser as soon as it is live, and checks for updates. A separate Update-AI-Toolkit script handles later upgrades.

Who it’s for

Windows users with NVIDIA GPUs who want to train or fine-tune image and video diffusion models without manually wrestling with package managers, CUDA versions, or virtual environments. It is aimed at both beginners and experienced users who prefer a portable, self-contained setup.

Highlights

  • One-click portable installer – no system Python, no VENV, just run the batch file.
  • Auto-provisions the whole stack – Git, Node.js, FFmpeg, Python, PyTorch, Triton, FlashAttention, torchcodec, and the AI-Toolkit itself.
  • Pinned compatible versions – FlashAttention 2.8.3, Triton 3.7.1.post27, torchcodec 0.15.0 for Torch 2.13.0 to avoid breakage.
  • Launcher convenience – opens the browser when the server is up and alerts you to updates.
  • Safe usage guidance – warns against running as Administrator, system folders, and spaces in paths.

Related

  • Project
  • Project
  • Project
  • Project
  • Project