torch==2.9.1
diffusers==0.33.0
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
zstandard
openpyxl
flake8~=7.1.1
flake8-json~=24.4.0
radon~=6.0.1
pylint~=3.3.3

[:platform_system != "Windows"]
litert-torch

[stat]
nn-stat
