尧图建网站 尧图建网站 YAOTU WEB BUILD 免费咨询
ARTICLE DETAIL

资讯详情

深耕网站建设与建站编程的一线实战洞察。

Transformers 实战:3 行代码跑通 pipeline 模型推理

Transformers 实战:3 行代码跑通 pipeline 模型推理 Transformers 实战3 行代码跑通 pipeline 模型推理【免费下载链接】transformers Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.项目地址: https://gitcode.com/GitHub_Trending/tra/transformers想上线一个大模型先得手写数据预处理、模型加载、输出解析等写完就劝退了。Transformers库把这些步骤压缩成一次pipeline调用指定任务和模型 ID结果直接给你。下面的示例全部复用仓库里现成的测试用例和样张。项目速览一句话定位文本、图像、音频模型的统一推理框架。能做什么一句话解释一行推理pipeline(task, model)自动完成预处理、前向与输出解析文本生成丢一段 prompt 进去拿回模型的续写图像分类传本地图片路径返回label/score列表语音识别音频进、转写文本出训练微调Trainer类加TrainingArguments驱动完整训练循环跑起来一行安装[torch]会带上 PyTorch 后端pip install transformers[torch]跑完下面的代码你会拿到一个包含generated_text的字典即模型对这句话的续写示例输出取自 README Quickstartfrom transformers import pipeline model pipeline(tasktext-generation, modelQwen/Qwen2.5-1.5B) model(the secret to baking a really good cake is )用 pipeline 提取图像分类结果 后端收到用户上传的图片需要知道里面有什么但不必自己写图像预处理。clf pipeline(taskimage-classification, modelfacebook/dinov2-small-imagenet1k-1-layer) clf(tests/fixtures/tests_samples/COCO/apple.jpg)tests/models/minimax_m3_vl/test_modeling_minimax_m3_vl.py 第 588 行的test_real_image_apple_recognition测试用例打开同一张苹果样张断言多模态模型的回答包含 apple图像分类这条管线用它可以直接验证。回答图片问题视觉问答 固定类别不够用、想用自然语言提问图片时换成视觉问答任务。vqa pipeline(taskvisual-question-answering, modelSalesforce/blip-vqa-base) vqa(imagetests/fixtures/tests_samples/COCO/000000004016.png, questionWhat is in the image?)VQA 入口pipeline()定义在 src/transformers/pipelines/init.py 里签名按task重载你填的任务决定预处理和输出结构。000000004016.png是 COCO 数据集的真实厨房场景图可直接替换验证调参 边界task指定推理入口模型与预处理由它自动匹配选错任务名会直接报错如image-classificationdevice指定运行设备如0或cpu多卡时直接传卡号dtype控制计算精度默认auto新卡上改torch.bfloat16更省显存 提示use_fast控制是否启用更快的 tokenizer 实现默认True个别旧模型不兼容时可切False。踩坑提醒首次运行会下载模型按 ID 拉取并缓存重复运行不再下载但首次耗时取决于网络。任务与模型要匹配task和model必须成对把视觉语言模型塞给纯文本任务会失败。示例输出可能不同模型权重与版本会更新README 的样例输出和你本地跑出的结果未必一致。训练定位不同README 明确说它不是模块化的神经网络工具箱examples/脚本是示例需要改造后使用。一个pipeline覆盖文本到图像的推理链路Trainer则接手微调环节更多任务用法见 README.md 的 Quickstart 章节。【免费下载链接】transformers Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.项目地址: https://gitcode.com/GitHub_Trending/tra/transformers创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
返回列表