diff --git a/README.md b/README.md index a003e61f433..55e6a3f9aec 100644 --- a/README.md +++ b/README.md @@ -134,7 +134,7 @@ from zenml.config import ResourceSettings, DockerSettings @step( settings={ - "resources": ResourceSettings(memory="16Gb", gpu="1", cpu="8"), + "resources": ResourceSettings(memory="16GB", gpu_count="1", cpu_count="8"), "docker": DockerSettings(parent_image="pytorch/pytorch:1.12.1-cuda11.3-cudnn8-runtime") } ) @@ -147,6 +147,9 @@ zenml stack set k8s # Set a stack with kubernetes orchestrator python run.py ``` +![Workloads with ZenML](/docs/book/.gitbook/assets/readme_compute.gif) + + ### Track models, pipeline, and artifacts Create a complete lineage of who, where, and what data and models are produced. diff --git a/docs/book/.gitbook/assets/readme_compute.gif b/docs/book/.gitbook/assets/readme_compute.gif new file mode 100644 index 00000000000..e210fceaefd Binary files /dev/null and b/docs/book/.gitbook/assets/readme_compute.gif differ diff --git a/docs/book/.gitbook/assets/readme_youtube_thumbnail.png b/docs/book/.gitbook/assets/readme_youtube_thumbnail.png index 970aaf527a4..34063c06cca 100644 Binary files a/docs/book/.gitbook/assets/readme_youtube_thumbnail.png and b/docs/book/.gitbook/assets/readme_youtube_thumbnail.png differ