requirements.txt

# TensorRT Diffusion Model Optimization Pipeline
# Core dependencies for NVIDIA T4 GPU optimization on Google Colab
#
# NOTE: On Google Colab, PyTorch 2.7.x is built against CUDA 12.6.
# torch-tensorrt must be pinned to match PyTorch version to avoid ABI mismatch.
#
# Google Colab install (PyTorch 2.7.x + CUDA 12.6):
#   pip install torch-tensorrt==2.7.0 --extra-index-url https://download.pytorch.org/whl/cu128
#   pip install tensorrt tensorrt-lean
#   pip install nvidia-modelopt>=0.15.0
#
# For local development with CUDA 12.8+:
#   pip install torch-tensorrt --extra-index-url https://download.pytorch.org/whl/cu128
#   pip install tensorrt nvidia-modelopt>=0.15.0

# NVIDIA Model Optimizer for INT8 PTQ
nvidia-modelopt>=0.15.0

# Diffusion model support
diffusers>=0.25.0
transformers>=4.36.0
accelerate>=0.25.0

# Image processing
Pillow>=9.0.0

# Numerical computing
numpy>=1.26.0

# Development dependencies
pytest>=7.0
hypothesis>=6.0
pytest-benchmark>=4.0