Fix typo

vllm-project · Feb 11, 2025 · e9be0ec · e9be0ec
1 parent 65299fa
commit e9be0ec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmarks/benchmark_latency.py b/benchmarks/benchmark_latency.py
@@ -44,7 +44,7 @@ def save_to_pytorch_benchmark_format(
         },
     }
 
-    if os.environ.get("SAVE_IN_PYTORCH_BENCHMARK_FORMAT", False):
+    if os.environ.get("SAVE_TO_PYTORCH_BENCHMARK_FORMAT", False):
         # Don't use json suffix here as we don't want convert-results-json-to-markdown.py
         # to pick it up
         output_file = f"{os.path.splitext(args.output_json)[0]}.pytorch"