From f559d6664c976a997f2a8604c44a79605927cd8f Mon Sep 17 00:00:00 2001 From: Wei Lin Date: Wed, 4 Dec 2024 17:04:33 +0800 Subject: [PATCH] Adjust Qwen2-7B test case (#1551) --- tests/test_text_generation_example.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_text_generation_example.py b/tests/test_text_generation_example.py index 1fcadba9b0..6b498c9313 100644 --- a/tests/test_text_generation_example.py +++ b/tests/test_text_generation_example.py @@ -51,7 +51,7 @@ ("google/gemma-2-9b", 1, False, 92.302359446567, True), ("state-spaces/mamba-130m-hf", 1536, False, 5385.511100161605, False), ("Deci/DeciLM-7B", 1, False, 120, False), - ("Qwen/Qwen2-7B", 512, False, 9669.45787, True), + ("Qwen/Qwen2-7B", 256, False, 8870.945160540245, True), ("Qwen/Qwen1.5-MoE-A2.7B", 1, True, 44.25834541569395, False), ("EleutherAI/gpt-neo-2.7B", 1, False, 257.2476416844122, False), ("facebook/xglm-1.7B", 1, False, 357.46365062825083, False), @@ -117,7 +117,7 @@ "meta-llama/Llama-2-7b-hf": "DeepSpeed is a machine learning framework for deep learning. It is designed to be fast and efficient, while also being easy to use. DeepSpeed is based on the TensorFlow framework, and it uses the TensorFlow library to perform computations.\nDeepSpeed is a deep learning framework that is designed to be fast and efficient. It is based on the TensorFlow library and uses the TensorFlow library to perform computations. DeepSpeed is designed to be easy to use and to provide a high level of flex", "mistralai/Mistral-7B-v0.1": "DeepSpeed is a machine learning framework that accelerates training of large models on a single machine or distributed systems. It is designed to be compatible with PyTorch and TensorFlow, and can be used to train models on a single machine or on a distributed system.\n\nDeepSpeed is a machine learning framework that accelerates training of large models on a single machine or distributed systems. It is designed to be compatible with PyTorch and TensorFlow, and can be used to train models on a single machine or on a distributed system", "mistralai/Mixtral-8x7B-v0.1": "DeepSpeed is a machine learning framework that enables training of large models on a single machine with a single GPU. It is designed to be easy to use and efficient, and it can be used to train models on a variety of tasks.\n\n## Introduction\n\nDeepSpeed is a machine learning framework that enables training of large models on a single machine with a single GPU. It is designed to be easy to use and efficient, and it can be used to train models on a variety of tasks.\n\n## What is DeepSpeed", - "Qwen/Qwen2-7B": "DeepSpeed is a machine learning framework that provides a suite of toolskits for building and training deep learning models. It is designed to be highly scalable and efficient, and it supports a wide range of deep learning frameworks, including PyTorch, TensorFlow, and MXNet. DeepSpeed is particularly well-suited for training large-scale models on distributed systems, and it provides a number of features that make it easy to use and configure. Some of the key features of DeepSpeed include:\n\n- Distributed training: DeepSpeed supports distributed training on multiple", + "Qwen/Qwen2-7B": "DeepSpeed is a machine learning framework that provides a unified interface for training deep learning models. It is designed to be easy to use and to provide high performance. DeepSpeed is built on top of PyTorch and TensorFlow, and it supports a wide range of models, including transformers, convolutional neural networks, and recurrent neural networks.\nDeepSpeed is a machine learning framework that provides a unified interface for training deep learning models. It is designed to be easy to use and to provide high performance. DeepSpeed is built on top of Py", } else: # Gaudi1 CI baselines