add test and merge release

This commit is contained in:
Zhicheng Zhang
2023-03-18 16:31:44 +08:00
parent 148a43d526
commit d117f2c007

View File

@@ -0,0 +1,29 @@
# Copyright (c) Alibaba, Inc. and its affiliates.
import unittest
from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks
from modelscope.utils.demo_utils import DemoCompatibilityCheck
from modelscope.utils.test_utils import test_level
class TextGenerationTest(unittest.TestCase, DemoCompatibilityCheck):
def setUp(self) -> None:
self.chatglm6b_model_id = 'ZhipuAI/ChatGLM-6B'
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
def test_chatglm6b(self):
pipe = pipeline(
task=Tasks.text_generation,
model=self.chatglm6b_model_id,
model_revision='v1.0.1')
text = '你好'
history = []
inputs = {'text': text, 'history': history}
result = pipe(inputs)
print(result)
if __name__ == '__main__':
unittest.main()