replacebg / requirements.txt
Munaf1987's picture
Update requirements.txt
4c3888a verified
# Core ML frameworks for professional AI generation
torch>=2.4.0
torchvision>=0.19.0
torchaudio>=2.4.0
transformers>=4.45.0
diffusers>=0.30.0
accelerate>=0.34.0
safetensors>=0.4.0
# FLUX and LoRA support (ZeroGPU optimized)
xformers>=0.0.27
peft>=0.7.0
# flash-attn>=2.6.0 # Commented out - requires CUDA compilation
# bitsandbytes>=0.41.0 # Commented out - can cause issues in Spaces
# Professional image and video processing
Pillow>=10.4.0
opencv-python>=4.9.0
numpy>=1.26.0
matplotlib>=3.8.0
scikit-image>=0.22.0
moviepy>=1.0.3
imageio>=2.31.0
imageio-ffmpeg>=0.4.9
ffmpeg-python>=0.2.0
# Audio processing (open source)
soundfile>=0.12.1
librosa>=0.10.1
pydub>=0.25.1
# Professional Gradio interface
gradio>=4.40.0
spaces>=0.28.0
# Enhanced utilities
requests>=2.31.0
tqdm>=4.66.0
huggingface-hub>=0.25.0
aiohttp>=3.9.0
websockets>=12.0
psutil>=5.9.0
rich>=13.7.0
# Advanced text processing
nltk>=3.8.0
textblob>=0.17.0
regex>=2023.10.0
# Performance optimization (ZeroGPU compatible)
einops>=0.7.0
omegaconf>=2.3.0
timm>=0.9.0
# triton>=2.1.0 # Commented out - requires CUDA compilation
# API support
python-multipart>=0.0.6
uvicorn>=0.24.0
fastapi>=0.104.0
pydantic>=2.4.0
typing-extensions>=4.8.0
# Async operations
aiofiles>=23.2.0
# Memory and GPU monitoring (optional)
# py3nvml>=0.2.7 # Commented out - requires NVIDIA drivers
# GPUtil>=1.4.0 # Commented out - requires NVIDIA drivers