torch==2.9.1
diffusers==0.33.0
#
# NOTE (Windows): `litert-torch` (successor to `ai-edge-torch`) pulls `tf-nightly`/
# `ai-edge-litert` which don't ship Windows wheels, so pip can't resolve them.
# The codebase already falls back to `tf.lite.Interpreter` when LiteRT isn't present.
# Install `litert-torch` only on non-Windows platforms.
litert-torch; platform_system != "Windows"
torchvision==0.24.1
transformers
bitsandbytes
accelerate
einops
ftfy
tqdm
onnx
onnxruntime
optuna
datasets>=3.2.0
kagglehub
pycocotools~=2.0.8
numpy
pandas
scipy
scikit-learn
matplotlib
requests>=2.32.3
pillow
setuptools
nltk>=3.9.1
evaluate
gdown
datasketch # to check code novelty
zstandard
openpyxl

# For the code evaluator
flake8~=7.1.1
flake8-json~=24.4.0
radon~=6.0.1
pylint~=3.3.3
