Buckets:

linoyts's picture
|
download
raw
6.71 kB

Quick Start Guide

Get up and running with LTX-2 training in just a few steps!

๐Ÿ“‹ Prerequisites

Before you begin, ensure you have:

  1. LTX-2 Model Checkpoint - A local .safetensors file containing the LTX-2 model weights. Download ltx-2.3-22b-dev.safetensors from: HuggingFace Hub
  2. Gemma Text Encoder - A local directory containing the Gemma model (required for LTX-2). Download from: HuggingFace Hub
  3. Linux with CUDA - The trainer requires triton which is Linux-only
  4. GPU with sufficient VRAM - 80GB recommended for the standard config. For GPUs with 32GB VRAM (e.g., RTX 5090), use the low VRAM config which enables INT8 quantization and other memory optimizations

โšก Installation

First, install uv if you haven't already. Then clone the repository and install the dependencies:

git clone https://github.com/Lightricks/LTX-2

The ltx-trainer package is part of the LTX-2 monorepo. Install the dependencies from the repository root, then navigate to the trainer package:

# From the repository root
uv sync
cd packages/ltx-trainer

The trainer depends on ltx-core and ltx-pipelines packages which are automatically installed from the monorepo.

๐Ÿ‹ Training Workflow

1. Prepare Your Dataset

Organize your videos and captions, then preprocess them:

# Split long videos into scenes (optional)
uv run python scripts/split_scenes.py input.mp4 scenes_output_dir/ --filter-shorter-than 5s

# Generate captions for videos (optional)
uv run python scripts/caption_videos.py scenes_output_dir/ --output dataset.json

# Preprocess the dataset (compute latents and embeddings)
uv run python scripts/process_dataset.py dataset.json \
    --resolution-buckets "960x544x49" \
    --model-path /path/to/ltx-2-model.safetensors \
    --text-encoder-path /path/to/gemma-model

See Dataset Preparation for detailed instructions.

2. Configure Training

Create or modify a configuration YAML file. Start with one of the example configs:

Key settings to update:

model:
  model_path: "/path/to/ltx-2-model.safetensors"
  text_encoder_path: "/path/to/gemma-model"

data:
  preprocessed_data_root: "/path/to/preprocessed/data"

output_dir: "outputs/my_training_run"

See Configuration Reference for all available options.

3. Start Training

uv run python scripts/train.py configs/t2v_lora.yaml

For multi-GPU training:

uv run accelerate launch scripts/train.py configs/t2v_lora.yaml

See Training Guide for distributed training and advanced options.

๐ŸŽฏ Training Modes

First time? Start with t2v_lora.yaml โ€” it's the simplest mode and only requires videos with captions. You can explore other modes once you've confirmed your setup works.

The trainer supports several training modes:

Mode Description Example Config
Text-to-Video Generate video+audio from text prompts t2v_lora.yaml
Image-to-Video Animate from a starting image i2v_lora.yaml
Video Extension Extend videos temporally (forward/backward) video_extend_lora.yaml
IC-LoRA (V2V) Video-to-video transformations v2v_ic_lora.yaml
Audio-to-Video Generate video conditioned on audio a2v_lora.yaml
Video-to-Audio Generate audio/foley from video v2a_lora.yaml
Video Inpainting Fill in masked regions of video video_inpainting_lora.yaml
Video Outpainting Extend video spatially video_outpainting_lora.yaml
Text-to-Audio Generate audio from text prompts t2a_lora.yaml
Audio Extension Extend audio temporally audio_extend_lora.yaml
Audio Inpainting Fill in masked regions of audio audio_inpainting_lora.yaml
IC-LoRA (A2A) Audio-to-audio transformations a2a_ic_lora.yaml
AV2AV IC-LoRA Audio+video IC-LoRA transformations av2av_ic_lora.yaml
Full Fine-tuning Full model training (any mode above) Set model.training_mode: "full"

See Training Modes for detailed explanations of each mode.

Next Steps

Once you've completed your first training run, you can:

Need Help?

If you run into issues at any step, see the Troubleshooting Guide for solutions to common problems.

Join our Discord community for real-time help and discussion!

Xet Storage Details

Size:
6.71 kB
ยท
Xet hash:
99b015a993e05233b2c97b653a90a7fdd990b8f3bd9078d31a1848bdaeb4184a

Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.