# ADOBE CONFIDENTIAL
# Copyright 2026 Adobe
# All Rights Reserved.
# NOTICE: All information contained herein is, and remains
# the property of Adobe and its suppliers, if any. The intellectual
# and technical concepts contained herein are proprietary to Adobe
# and its suppliers and are protected by all applicable intellectual
# property laws, including trade secret and copyright laws.
# Dissemination of this information or reproduction of this material
# is strictly forbidden unless prior written permission is obtained
# from Adobe.

# Core runtime for TPIPS (training + evaluation + retrieval demo run from the
# repo checkout). Install with: pip install -r requirements.txt
# torch is pulled from PyPI (a CUDA build by default); install a specific CUDA
# build from https://pytorch.org/get-started/locally/ first if you need one.
# torch>=2.7: transformers>=5.0 imports torch.float8_e8m0fnu (added in torch 2.7)
# at import time, so older torch crashes on `import transformers`.
# torchvision: the Qwen3-VL processor pulls in a video sub-processor that needs
# it. Install it from the SAME CUDA channel as torch (e.g. .../whl/cu126).
torch>=2.7
torchvision>=0.22
transformers>=5.0
accelerate>=1.10
peft>=0.19
qwen-vl-utils
omegaconf>=2.3
pillow>=10
numpy>=1.26
tqdm>=4.66
huggingface_hub>=0.26
hf_xet>=1.0              # Xet chunked-transfer backend (fast up/download)

# flash-attn is optional. TPIPS automatically uses compatible FlashAttention
# when available and otherwise uses PyTorch SDPA. The helper is optional:
#     bash scripts/install_flash_attn.sh
