add test.

Signed-off-by: Ye, Xinyu <[email protected]>
XinyuYe-Intel · Sep 10, 2024 · c48fc46 · c48fc46
1 parent 2f78787
commit c48fc46
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_text_generation_example.py b/tests/test_text_generation_example.py
@@ -42,6 +42,7 @@
             ("google/gemma-7b", 1, False, 109.70751574382221),
             ("state-spaces/mamba-130m-hf", 1536, False, 5385.511100161605),
             ("Deci/DeciLM-7B", 1, False, 120),
+            ("facebook/xglm-1.7B", 1, False, 357.46365062825083),
         ],
         "fp8": [
             ("tiiuae/falcon-180B", 4, 950, True, 128, 128, 2506.68),