wehub-resource-sync
23 lines
428 B
YAML
23 lines
428 B
YAML
# See https://hub.docker.com/r/anyscale/ray for full list of
|
|
# available Ray, Python, and CUDA versions.
|
|
base_image: anyscale/ray:2.9.0-py39-cu118
|
|
|
|
env_vars: {}
|
|
|
|
debian_packages: []
|
|
|
|
python:
|
|
pip_packages:
|
|
- accelerate==0.20.3
|
|
- diffusers==0.17.1
|
|
- fastapi==0.97.0
|
|
- ipywidgets
|
|
- matplotlib==3.7.1
|
|
- numpy==1.24.3
|
|
- torch==2.0.1
|
|
- transformers==4.30.1
|
|
|
|
conda_packages: []
|
|
|
|
post_build_cmds: []
|