[no ci] Pin torchao and fbgemm-gpu-genai dependencies (#115)

huydhn · web-flow · commit a0694153be8c · 2025-12-15T11:17:12.000-08:00
Signed-off-by: Huy Do &lt;huydhn@gmail.com&gt;
diff --git a/.github/workflows/vllm-benchmark.yml b/.github/workflows/vllm-benchmark.yml
@@ -336,7 +336,7 @@ jobs:
             "${DOCKER_IMAGE}"
           )
           if [[ "${DEVICE_NAME}" == "cuda" ]]; then
-            docker exec -t "${container_name}" bash -c "pip install torchao fbgemm-gpu-genai"
+            docker exec -t "${container_name}" bash -c "pip install torchao==0.14.1 fbgemm-gpu-genai==1.4.1"
           fi
           docker exec -t "${container_name}" bash -c "cd vllm-benchmarks/vllm && bash .buildkite/performance-benchmarks/scripts/run-performance-benchmarks.sh"
 

Original file line number	Diff line number	Diff line change
`@@ -336,7 +336,7 @@ jobs:`
`336`	`336`	`"${DOCKER_IMAGE}"`
`337`	`337`	`)`
`338`	`338`	`if [[ "${DEVICE_NAME}" == "cuda" ]]; then`
`339`		`- docker exec -t "${container_name}" bash -c "pip install torchao fbgemm-gpu-genai"`
	`339`	`+ docker exec -t "${container_name}" bash -c "pip install torchao==0.14.1 fbgemm-gpu-genai==1.4.1"`
`340`	`340`	`fi`
`341`	`341`	`docker exec -t "${container_name}" bash -c "cd vllm-benchmarks/vllm && bash .buildkite/performance-benchmarks/scripts/run-performance-benchmarks.sh"`
`342`	`342`