[CI] update the falcon model for TRTLLM (#1853)

deepjavalibrary · May 1, 2024 · 6a3122a · 6a3122a
1 parent 31a3a74
commit 6a3122a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/integration/llm/prepare.py b/tests/integration/llm/prepare.py
@@ -502,7 +502,7 @@
         "option.output_formatter": "jsonlines",
     },
     "falcon-7b": {
-        "option.model_id": "s3://djl-llm/triton/0.8.0/falcon-7b-tp1-bs16/",
+        "option.model_id": "s3://djl-llm/triton/0.9.0/falcon-7b-tp1-bs16/",
         "option.tensor_parallel_degree": 1,
         "option.max_input_len": 1024,
         "option.max_output_len": 512,