diff --git a/distributions/meta-reference-gpu/compose.yaml b/distributions/meta-reference-gpu/compose.yaml index 70b37f260..2b88c68fc 100644 --- a/distributions/meta-reference-gpu/compose.yaml +++ b/distributions/meta-reference-gpu/compose.yaml @@ -25,11 +25,10 @@ services: # satisfy all the requested capabilities for a successful # reservation. capabilities: [gpu] - runtime: nvidia - entrypoint: bash -c "python -m llama_stack.distribution.server.server --yaml_config /root/my-run.yaml" - deploy: restart_policy: condition: on-failure delay: 3s max_attempts: 5 window: 60s + runtime: nvidia + entrypoint: bash -c "python -m llama_stack.distribution.server.server --yaml_config /root/my-run.yaml"