Skip to content

GitLab

"figs/.gitkeep" did not exist on "bcdfbfd57c8f3cd6cd65998464bb71a562d49948"
Unverified Commit 30ee3630 authored by Lianmin Zheng's avatar Lianmin Zheng Committed by GitHub
Browse files

Fix the failed unit tests (#1699)

parent e5db40dc
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
test/srt/models/test_generation_models.py
View file @ 30ee3630
......@@ -157,6 +157,7 @@ class TestGenerationModels(unittest.TestCase):
if model_case.model_path in [
"HuggingFaceTB/SmolLM-135M-Instruct",
"allenai/OLMo-1B-0724-hf",
"google/gemma-2-2b", # There is a bug with new transformers library. This can only run with transformers==4.44
]:
prompts = [p for p in DEFAULT_PROMPTS if len(p) < 1000]
......
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