joycaption / requirements.txt
kazuhina's picture
Fix accelerate dependency and device_map compatibility for Spaces
e6d8d3d
raw
history blame contribute delete
399 Bytes
# JoyCaption - Required Dependencies for Spaces
# Core ML/AI libraries (not included in Spaces base)
transformers>=4.40.0
torch>=2.0.0
torchvision>=0.15.0
accelerate>=0.20.0 # Required for device_map
# Image processing
Pillow>=10.0.0
# Gradio and UI
gradio>=5.0.0
spaces>=0.19.0
# Model quantization support
bitsandbytes>=0.41.0
# Optional: Hugging Face Hub enhancements
huggingface_hub>=0.15.0