| # Common dependencies | |
| -r requirements-common.txt | |
| # Dependencies for NVIDIA GPUs | |
| ray >= 2.9 | |
| nvidia-ml-py # for pynvml package | |
| torch == 2.4.0 | |
| # These must be updated alongside torch | |
| torchvision == 0.19 # Required for phi3v processor. See https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version | |
| xformers == 0.0.27.post2; platform_system == 'Linux' and platform_machine == 'x86_64' # Requires PyTorch 2.4.0 | |