fix broken github action

huggingface · Aug 5, 2024 · 626f613 · 626f613
1 parent 6a4bf71
commit 626f613
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llm_perf/update_llm_perf_cuda_pytorch.py b/llm_perf/update_llm_perf_cuda_pytorch.py
@@ -134,7 +134,7 @@ def benchmark_cuda_pytorch(model, attn_implementation, weights_config):
         quantization_scheme=quant_scheme,
         quantization_config=quant_config,
         attn_implementation=attn_implementation,
-        hub_kwargs={"trust_remote_code": True},
+        model_kwargs={"trust_remote_code": True},
     )
 
     benchmark_config = BenchmarkConfig(