diff --git a/ai/orchestrators/benchmarking.mdx b/ai/orchestrators/benchmarking.mdx index 3e5051c9..62fcde26 100644 --- a/ai/orchestrators/benchmarking.mdx +++ b/ai/orchestrators/benchmarking.mdx @@ -47,12 +47,12 @@ Orchestrator node. Example command: ```bash - docker run --gpus 0 -v ./models:/models livepeer/ai-runner:latest python bench.py --pipeline text-to-image --model_id stabilityai/sd-turbo --runs 3 --batch_size 3 + docker run --gpus '"device=0"' -v ./models:/models livepeer/ai-runner:latest python bench.py --pipeline text-to-image --model_id stabilityai/sd-turbo --runs 3 --batch_size 3 ``` In this command, the following parameters are used: - - ``: Specify which GPU(s) to use. For example, `0` for GPU 0, `0,1` for GPU 0 and GPU 1, or `all` for all GPUs. + - ``: Specify which GPU(s) to use. For example, `'"device=0"'` for GPU 0, `'"device=0,1"'` for GPU 0 and GPU 1, or `'"device=all"'` for all GPUs. - ``: The pipeline to benchmark (e.g., `text-to-image`). - ``: The model ID to use for benchmarking (e.g., `stabilityai/sd-turbo`). - ``: The number of benchmark runs to perform.