Switcher / requirements.txt
crash10155's picture
Update requirements.txt
84ba5a8 verified
# === CORE ML/AI STACK ===
transformers==4.46.2
diffusers==0.31.0
accelerate>=0.20.0,<1.0.0
safetensors==0.4.5
tokenizers==0.20.3
huggingface-hub==0.32.5
# === WEB/API FRAMEWORKS ===
gradio==5.33.1
fastapi==0.115.12
uvicorn==0.34.3
# === COMPUTER VISION ===
opencv-python==4.10.0.84
pillow==11.2.1
albumentations==2.0.8
scikit-image==0.25.2
# === SCIENTIFIC/DATA SCIENCE ===
numpy==1.26.4
scipy==1.15.3
scikit-learn==1.7.0
pandas==2.2.2
matplotlib==3.9.2
onnx==1.18.0
onnxruntime-gpu==1.19.2
# === TRAINING/MONITORING ===
tensorflow==2.18.0
tensorboard==2.18.0
# === UTILITIES ===
requests==2.32.4
tqdm==4.67.1
psutil==7.0.0
moviepy==1.0.3
opennsfw2==0.14.0
--extra-index-url https://download.pytorch.org/whl/cu118
torch==2.0.1+cu118
torchvision==0.15.2+cu118
torchaudio==2.0.2+cu118
torchmetrics==1.7.2
insightface==0.7.3
realesrgan==0.3.0
gfpgan==1.3.8
basicsr==1.4.2
librosa==0.10.2
soundfile==0.12.1
ffmpeg-python==0.2.0
pydub==0.25.1
imageio==2.34.1