# Pinned dependencies for the POPE audit and the inference-time
# correction experiments. Tested on Python 3.10 + CUDA 11.8.
#
# Install with:
#   pip install -r requirements.txt
#
# Torch is intentionally not pinned to a CUDA wheel here; install it
# from https://pytorch.org/get-started/locally/ with the index URL that
# matches your CUDA version, then run pip install -r requirements.txt.
# Example:
#   pip install torch==2.2.* torchvision==0.17.* torchaudio==2.2.* \
#     --index-url https://download.pytorch.org/whl/cu118

# Core ML
transformers==4.40.1
tokenizers>=0.15,<0.20
accelerate>=0.28,<0.30
bitsandbytes>=0.43,<0.44
sentencepiece>=0.2.0
protobuf>=3.20,<5.0
huggingface_hub>=0.22,<0.25
safetensors>=0.4.2

# Datasets + image
datasets>=2.18,<3.0
Pillow>=10.0,<12.0
requests>=2.31

# Numerics + plots
numpy>=1.24,<2.0
pandas>=2.0,<3.0
scipy>=1.10,<2.0
scikit-learn>=1.3,<2.0
matplotlib>=3.7,<4.0
seaborn>=0.13,<0.15
tqdm>=4.66

# Notebook + utility (optional for figure regeneration)
jupyter>=1.0
