SceneWeaver / requirements.txt
DawnC's picture
Upload 10 files
ca80d1d verified
raw
history blame
2.73 kB
# SceneWeaver Hugging Face Spaces Deployment Requirements
# Optimized for ZeroGPU environment with safe version ranges
# ============================================
# Core Deep Learning Framework
# ============================================
# PyTorch 2.x series - compatible with SDXL and xformers
torch>=2.0.0,<2.5.0
torchvision>=0.15.0,<0.20.0
torchaudio>=2.0.0,<2.5.0
# ============================================
# Diffusion Models and Transformers
# ============================================
# Diffusers 0.25+ has better SDXL support, <0.32 for stability
diffusers>=0.25.0,<0.32.0
# Transformers compatible with diffusers and open_clip
transformers>=4.35.0,<4.46.0
# Accelerate for model loading optimizations
accelerate>=0.24.0,<0.35.0
# xformers for memory efficient attention (optional, may fail on some systems)
# xformers>=0.0.22,<0.0.29
# ============================================
# Computer Vision and Image Processing
# ============================================
# OpenCV for image processing
opencv-python>=4.8.0,<4.11.0
# opencv-contrib-python for guided filter (cv2.ximgproc)
opencv-contrib-python>=4.8.0,<4.11.0
# Pillow for image I/O
Pillow>=9.5.0,<11.0.0
# SciPy for scientific computing
scipy>=1.10.0,<1.15.0
# ============================================
# Background Removal
# ============================================
# rembg for foreground extraction (CPU version for compatibility)
rembg>=2.0.50,<2.1.0
# ============================================
# Multi-modal Understanding (CLIP)
# ============================================
# OpenCLIP for image analysis
open_clip_torch>=2.20.0,<2.27.0
# Sentence transformers (dependency)
sentence-transformers>=2.2.0,<3.1.0
# ============================================
# Web Interface
# ============================================
# Gradio 4.x for modern UI
gradio>=4.0.0,<5.0.0
# ============================================
# Core Scientific Computing
# ============================================
# NumPy 1.x for compatibility
numpy>=1.24.0,<2.0.0
# ============================================
# Hugging Face Integration
# ============================================
# HuggingFace Hub for model downloads
huggingface_hub>=0.19.0,<0.27.0
# Safetensors for efficient model loading
safetensors>=0.4.0,<0.5.0
# ============================================
# System Utilities
# ============================================
# psutil for memory monitoring
psutil>=5.9.0,<6.1.0
# requests for HTTP operations
requests>=2.28.0,<2.33.0
# ============================================
# Hugging Face Spaces (auto-installed on Spaces)
# ============================================
# spaces # ZeroGPU support - auto-available on HF Spaces