# MiniMax-H3 inference (infer.py / convert.py)
# Versions pinned to the working docker env (CUDA 12.6).
#
# Install torch first from the CUDA wheel index, then the rest:
#   pip install torch==2.9.1 torchvision==0.24.1 --index-url https://download.pytorch.org/whl/cu126
#   pip install -r requirements.txt
#
# MiniMax-H3 is not in a released diffusers build yet — install the local checkout
# (or the PR) instead of the PyPI package:
#   pip install -e ./diffusers
#   # or: pip install git+https://github.com/huggingface/diffusers.git@refs/pull/14355/head

# --- PyTorch (install via the CUDA index above; listed here for reference) ---
# torch==2.9.1+cu126
# torchvision==0.24.1+cu126

torch==2.9.1
torchvision==0.24.1

# --- Hugging Face stack ---
transformers==5.14.1
accelerate==1.14.0
huggingface-hub==1.26.0
safetensors==0.8.0
peft==0.20.0
tokenizers==0.22.2
sentencepiece==0.2.0
protobuf==4.24.4

# --- Diffusers extras used by infer.py ---
kernels==0.16.0

# --- Numerics / IO ---
numpy==1.24.4
scipy==1.12.0
Pillow==10.2.0
einops==0.7.0
ftfy==6.2.0
opencv-python==4.7.0
av==17.1.0
soundfile==0.12.1
imageio==2.37.4
imageio-ffmpeg==0.6.0