ai-parrot>=0.24.17
decorator>=5

[all]
ai-parrot-loaders[audio,document,ebook,ml,ml-heavy,pdf,video,web,youtube]

[audio]
whisperx==3.4.2
av==16.1.0
resemblyzer==0.1.4
pyannote-audio==3.4.0
pyannote-core==5.0.0
pyannote-database==5.1.3
pyannote-metrics==3.2.1
pyannote-pipeline==3.0.1
torch-audiomentations==0.12.0
torch-pitch-shift==1.2.5
torchmetrics==1.8.2

[document]
mammoth>=1.11.0

[ebook]
ebooklib>=0.19

[ml]
torch==2.6.0
torchaudio==2.6.0
torchvision==0.21.0
pytorch-lightning==2.5.5
pytorch-metric-learning==2.9.0
nvidia-cudnn-cu12==9.1.0.70

[ml-heavy]
torch==2.6.0
torchaudio==2.6.0
numpy<2.2,>=2.1
accelerate>=1.1.0
bitsandbytes==0.49.2
datasets>=3.0.2
transformers<=4.51.3,>=4.51.1
tensorflow>=2.19.1
tf-keras==2.19.0
opencv-python==4.10.0.84

[pdf]
paddleocr==3.2.0

[video]
moviepy==2.2.1
ffmpeg==1.4

[web]
beautifulsoup4>=4.12
html2text>=2024.0

[youtube]
pytube==15.0.0
youtube_transcript_api==1.0.3
yt-dlp>=2026.02.21
