torch>=2.0
torchvision>=0.15
diffusers>=0.30
transformers>=4.30
controlnet_aux>=0.0.6
compel>=2.0
omegaconf>=2.3
hydra-core>=1.3
accelerate>=0.20
numpy>=1.24
pillow>=9.0
opencv-python>=4.5
scipy>=1.10
tqdm
pyyaml
openai>=1.0

[all]
ciagen[captioning,datasets,examples,training]

[captioning]
ollama>=0.1

[datasets]
pycocotools
kaggle>=1.5
pandas>=2.0
datasets>=2.0

[dev]
pytest>=7.0
ruff>=0.1
pre-commit>=3.0

[docs]
mkdocs-material>=9.0
mkdocstrings[python]>=0.20
pymdown-extensions>=10.0

[examples]
matplotlib>=3.5
seaborn
jupyter
ipykernel

[training]
ultralytics>=8.0
wandb>=0.15
pyiqa>=0.1
