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

资讯详情

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

还在为 API 烧钱?我把 DeepSeek-R1 塞进浏览器本地跑,3 步搞定推理,附 5 个踩坑实录

还在为 API 烧钱?我把 DeepSeek-R1 塞进浏览器本地跑,3 步搞定推理,附 5 个踩坑实录 「你这个聊天窗口怎么不卡AI 推理不是都得放服务器上吗」同事看我演示完本地 DeepSeek 推理整个人愣住了。我告诉他没有服务器没有 API 调用数据连你的电脑都没出过。这篇文章你能得到什么零成本在浏览器里跑起 DeepSeek-R11.5B 量化版推理用WebGPU调用显卡加速不依赖云端用Web Worker隔离重计算页面永不卡死用单例模式让 1GB 模型只加载一次我踩过的5 个坑全帮你提前踩平全文代码可直接运行跟着做你也能拥有一个纯本地、可离线的 AI 聊天应用。 为什么我非要在浏览器里跑大模型先说我之前的痛调 API按 token 收费对话一多钱包就疼调 API网络一抖就超时生成还要等服务器排队调 API数据得发到别人服务器敏感内容没法聊本地部署要显卡、要 CUDA、要配环境直接劝退。直到我发现一条新路大模型 → 浏览器本地 → WebGPU 推理。零服务器、零 API 费用数据不出浏览器天然隐私加载一次后可离线使用推理跑在你的 GPU 上速度比想象中快这就是我做的webgpu-deepseek项目一个纯浏览器端的 DeepSeek-R1 聊天应用。 先搞懂数据流模型是怎么跑进浏览器的一句话流程HuggingFace 模型仓库 → transformers.jsJS 版 Transformers → 浏览器下载模型文件 → 浏览器缓存下次免下载 → WebGPU调用 GPU 加速 → 本地推理输出结果几个关键角色HuggingFaceAI 圈最火的开源模型社区各家模型都发在这里transformers.jsJS 版本的 transformers 库负责加载模型、执行推理WebGPU浏览器新特性让前端能直接调用 GPU我选的是DeepSeek-R1-Distill-Qwen-1.5B1.5B 参数量化后约1GB是目前浏览器端性价比最高的推理模型之一。 开工装依赖 搭架构第一步装两个依赖npmi huggingface/transformersnpmi markedhuggingface/transformers加载模型 执行推理marked模型输出的是Markdown得先转成 HTML 才能展示第二步想清楚架构推理是重计算直接跑在主线程页面必卡死。所以用Web Worker把推理隔离出去主线程只负责 UI主线程React UI ↕ postMessage 通信 Web Workerwork.js加载模型 推理Worker 和主线程之间用postMessage收发消息协议就五个动作switch(type){casecheck:check();break;// 检测 WebGPUcaseload:load();break;// 加载模型casegenerate:stopping_criteria.reset();generate(data);break;// 推理caseinterrupt:stopping_criteria.interrupt();break;// 停止生成casereset:past_key_values_cachenull;stopping_criteria.reset();break;// 重置} 单例模式让 1GB 模型只加载一次这是全文我最想讲的设计模式。单例模式OOP 面向对象里的 23 种经典设计模式之一核心就一句话——一个类在系统中只能实例化一次全局只有这一个实例。它专门解决两件事全局变量问题instance 到处传太痛苦全局状态问题状态要全局唯一共享放到大模型场景价值直接拉满1GB 的模型加载一次要几秒甚至几分钟。每次提问都重新加载直接劝退。单例模式保证整个页面生命周期模型只加载一次之后一直复用。看代码就在work.js里classTextGenerationPipeline{staticmodel_idonnx-community/DeepSeek-R1-Distill-Qwen-1.5B-ONNX;staticasyncgetInstance(progress_callbacknull){this.tokenizer??AutoTokenizer.from_pretrained(this.model_id,{progress_callback,});this.model??AutoModelForCausalLM.from_pretrained(this.model_id,{dtype:q4f16,device:webgpu,progress_callback,});returnPromise.all([this.tokenizer,this.model]);}}注意??空值合并赋值。第一次调用实例是空的走加载逻辑以后每次调用实例已存在直接返回懒加载 全局唯一一次到位。q4f16是量化精度device: webgpu指定走 GPU。 流式输出 R1 的思考过程大模型推理不能干等要边生成边吐字体验才对。用TextStreamer实现流式输出conststreamernewTextStreamer(tokenizer,{skip_prompt:true,skip_special_tokens:true,callback_function,// 每生成一段发给主线程token_callback_function,// 每个 token 回调统计速度});R1 还有个灵魂设计思考过程。它会先输出think.../think思考再输出正式回答。用两个特殊 token 做状态机// 151648: think// 151649: /thinkconst[START_THINKING_TOKEN_ID,END_THINKING_TOKEN_ID]tokenizer.encode(think/think,{add_special_tokens:false},);letstatethinking;// thinking or answeringconsttoken_callback_function(tokens){if(tokens[0]END_THINKING_TOKEN_ID){stateanswering;}};主线程拿到state就能把「思考」和「回答」分开展示还能实时算速度tokens/秒。生成时限制max_new_tokens: 2048并用InterruptableStoppingCriteria支持随时打断。 我踩的 5 个坑重点坑 1navigator.gpu 报错TS 不认识 WebGPUconstIS_WEBGPU_AVAILABLE!!navigator.gpu;一编译就报错Property gpu does not exist on type Navigator。原因WebGPU 是太新的实验特性TypeScript 自带类型里还没有它。当时的应急写法是类型断言constIS_WEBGPU_AVAILABLE!!(navigatorasany).gpu;但不建议到处乱用as any会把类型系统全部架空。正确解法安装类型声明文件npmi-Dwebgpu/types然后在tsconfig.app.json的types里声明{compilerOptions:{types:[webgpu/types]}}本质TS 靠.d.ts类型声明文件工作缺啥补啥。坑 2WebGPU 兼容性不是所有浏览器都能跑WebGPU 目前Chrome 113 / Edge默认支持部分浏览器还得手动开 flag。所以启动前必须做特性检测asyncfunctioncheck(){constadapterawaitnavigator.gpu.requestAdapter();if(!adapter){thrownewError(WebGPU is not supported (no adapter found));}}不支持就直接黑屏提示别让用户一脸懵。坑 3模型 1GB首次下载慢到怀疑人生首次加载要把模型文件从 HuggingFace 下载到浏览器1GB 起步没进度条根本不敢等。解决进度回调progress_callback实时上报主线程渲染进度条浏览器缓存下载一次之后走缓存二次加载秒开AutoModelForCausalLM.from_pretrained(this.model_id,{dtype:q4f16,device:webgpu,progress_callback,// 上报文件下载进度});坑 4首轮推理慢到爆炸其实是 shader 编译模型加载完了第一次生成还是卡好久因为WebGPU 要现场编译 shader。解决加载完用 dummy 输入跑一次提前编译asyncfunctionload(){const[tokenizer,model]awaitTextGenerationPipeline.getInstance();// 用假输入跑一遍把 shader 提前编译好constinputstokenizer(a);awaitmodel.generate({...inputs,max_new_tokens:1});self.postMessage({status:ready});}warmup 一次之后推理就丝滑了。坑 5模型输出乱成一坨忘了转 Markdown模型返回的是 Markdown直接塞进textContent代码块、加粗全废。必须用marked转成 HTML 再渲染import{marked}frommarked;// 生成完成后把 markdown 转成 HTMLchat.innerHTMLmarked.parse(markdownText); 最后回头看在浏览器里跑大模型并没有想象中那么科幻transformers.js抹平了模型加载的复杂度WebGPU把 GPU 能力直接给到前端单例模式解决重资源重复加载问题Web Worker保证页面流畅剩下的就是踩坑适合的场景个人工具、离线应用、隐私敏感场景、不想为 API 付费的玩具。
返回列表