Commit 54b92ba4 authored by zhuwenwen's avatar zhuwenwen
Browse files

update tests

parent f8382145
...@@ -7,7 +7,7 @@ import os ...@@ -7,7 +7,7 @@ import os
from vllm import LLM, SamplingParams from vllm import LLM, SamplingParams
from .conftest import get_text_from_llm_generator from .conftest import get_text_from_llm_generator
from ...utils import models_path_prefix from ....utils import models_path_prefix
# relatively small model with 4k sliding window # relatively small model with 4k sliding window
MODEL = os.path.join(models_path_prefix, "bigcode/starcoder2-3b") MODEL = os.path.join(models_path_prefix, "bigcode/starcoder2-3b")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment