Pipeline API是Transformers最核心的功能之一,使用流程如下:
1. 文本生成示例:
from transformers import pipeline
generator = pipeline(task="text-generation", model="Qwen/Qwen2.5-1.5B")
result = generator("The secret to baking a really good cake is")
print(result[0]["generated_text"])
2. 语音识别示例:
asr = pipeline(task="automatic-speech-recognition", model="openai/whisper-large-v3")
result = asr("https://huggingface.co/datasets/Narsil/asr_dummy/resolve/main/mlk.flac")
print(result["text"])
Wichtige Punkte:
- 模型会自动下载并缓存到~/.cache/huggingface/hub
- 可以通过TRANSFORMERS_CACHE环境变量修改缓存路径
- 支持本地音频文件或URL作为输入
Diese Antwort stammt aus dem ArtikelTransformers: Open Source Framework für maschinelles Lernen mit Unterstützung für Text-, Bild- und multimodale AufgabenDie