# Core Dependencies - Cross-platform compatible
# Install PyTorch separately based on your platform and CUDA version:
#   Linux/Windows with CUDA 12.4: pip install torch --index-url https://download.pytorch.org/whl/cu124
#   Linux/Windows with CUDA 12.1: pip install torch --index-url https://download.pytorch.org/whl/cu121
#   macOS: pip install torch

# NVIDIA ModelOpt (requires PyTorch to be installed first)
nvidia-modelopt[all]>=0.27.0

# Model handling
safetensors>=0.3.0
accelerate>=0.20.0

# Hugging Face libraries
transformers>=4.30.0
diffusers>=0.20.0

# Optional but useful
onnx>=1.14.0
onnxruntime>=1.15.0

# Note: Triton is installed automatically with ModelOpt on supported platforms
# For Windows, use: pip install -r requirements-windows.txt
# For Linux, use: pip install -r requirements-linux.txt (if needed)
