fix deprecation warning

neuralmagic · Oct 8, 2024 · 51412c5 · 51412c5
1 parent dd773c0
commit 51412c5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/offline_profile.py b/examples/offline_profile.py
@@ -108,7 +108,7 @@ def add_requests():
 
             llm.llm_engine.add_request(
                 request_id=f"seq{i}",
-                inputs={'prompt_token_ids': prompt_token_ids},
+                prompt={'prompt_token_ids': prompt_token_ids},
                 params=sampling_params)
     def abort_requests():
         for i in range(batch_size):