PIWM / requirements.txt
musictimer's picture
Fix bug 8
062758f
raw
history blame contribute delete
948 Bytes
# Core ML dependencies
torch>=1.13.0
torchvision>=0.14.0
torchaudio>=0.13.0
numpy>=1.21.0,<2.0.0
# Configuration management
hydra-core>=1.2.0
omegaconf>=2.2.0
# Web framework for deployment
fastapi>=0.68.0
uvicorn>=0.15.0
websockets>=10.0
python-multipart>=0.0.5
requests>=2.25.0
# Image processing
opencv-python-headless>=4.5.0
Pillow>=8.0.0
pygame>=2.1.0
# Hugging Face integration
huggingface-hub>=0.10.0
# Data handling
h5py>=3.7.0
# Atari environment dependencies
ale_py>=0.8.0
gymnasium>=0.28.0
# Experiment tracking (optional, for training only)
# wandb>=0.13.0 # Commented out due to NumPy 2.0 compatibility issues
# Metrics (required by rew_end_model.py)
torcheval>=0.0.6
# Progress bars (may be used by various components)
tqdm>=4.64.0
# Optional: for better performance
# torch-audio # if needed for audio processing
# Development dependencies (uncomment for local development)
# pytest>=6.0.0
# black>=21.0.0
# isort>=5.0.0