simba-watermark-cleaner / requirements.txt
JustForWorld's picture
fix: Sync ultralytics version to match training environment
2f69386
raw
history blame contribute delete
869 Bytes
# ===============================
# Веб-сервер
# ===============================
fastapi==0.108.0
uvicorn[standard]==0.29.0
python-multipart==0.0.9
requests==2.32.3
# ===============================
# Детектор объектов (YOLOv8)
# ===============================
ultralytics==8.3.208
# ===============================
# Утилиты
# ===============================
numpy
loguru==0.7.2
opencv-python-headless==4.9.0.80
Pillow==9.5.0
einops
# ===============================
# --- Hugging Face Ecosystem ---
# ===============================
# Совместимые версии для SD2 Inpainting и torch 2.3+/CUDA 12.1
diffusers==0.30.0
transformers==4.37.2
accelerate==0.24.1
huggingface-hub==0.24.6
# ===============================
# Прочее (иногда нужно diffusers)
# ===============================
safetensors