frozenleaves/openmind
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

!7 将mindformers中的text_generation修改为text-genertation

Browse Source 
Merge pull request !7 from cqq_601399/dev
This commit is contained in:
cqq_601399
2024-11-26 07:31:25 +00:00
committed by i-robot
parent bc7b040cbf
commit 7f897d88c5
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
docs/zh/basic_tutorial/pipeline.md
View File

@ -117,7 +117,7 @@ openMind Library `pipeline`方法支持PyTorch和MindSpore两种框架。此外
ms.set_context(mode=0, device_id=0, jit_level='o0', infer_boost='on', max_device_memory='59GB')
pipe = pipeline(task="text_generation",
pipe = pipeline(task="text-generation",
model='MindSpore-Lab/qwen1_5_7b',
framework='ms',
model_kwargs={"use_past": True},
@ -168,7 +168,7 @@ openMind Library `pipeline`方法支持PyTorch和MindSpore两种框架。此外
```python
from openmind import pipeline
text_pipeline_ms = pipeline(task="text_generation", model="MindSpore-Lab/baichuan2_7b_chat", framework='ms')
text_pipeline_ms = pipeline(task="text-generation", model="MindSpore-Lab/baichuan2_7b_chat", framework='ms')
output = text_pipeline_ms("hello!")
```
@ -182,7 +182,7 @@ PyTorch框架支持以下两种`backend`：`transformers`和`diffusers`。MindSp
```python
from openmind import pipeline
text_pipeline_ms = pipeline(task="text_generation", model="AI-Research/Qwen2-7B", framework='ms', backend="mindnlp")
text_pipeline_ms = pipeline(task="text-generation", model="AI-Research/Qwen2-7B", framework='ms', backend="mindnlp")
output = text_pipeline_ms("Give me some advice on how to stay healthy.")
```