{"id":120,"date":"2024-07-22T16:13:19","date_gmt":"2024-07-22T08:13:19","guid":{"rendered":"https:\/\/www.onetts.com\/ai\/?p=120"},"modified":"2024-07-22T16:13:19","modified_gmt":"2024-07-22T08:13:19","slug":"yi-6b","status":"publish","type":"post","link":"https:\/\/www.onetts.com\/ai\/models\/yi-6b\/","title":{"rendered":"Yi-6B"},"content":{"rendered":"<p data-spm-anchor-id=\"a2c6h.13066512.0.i3.4aa336af7aKEaS\">Yi-6B \u662f\u7531 01.AI \u96f6\u4e00\u4e07\u7269\u5f00\u53d1\u7684\u4e00\u7cfb\u5217\u5927\u578b\u8bed\u8a00\u6a21\u578b\uff08LLMs\uff09\u4e4b\u4e00\u3002\u8fd9\u4e00\u7cfb\u5217\u6a21\u578b\u4ee5\u5176\u5f00\u6e90\u6027\u548c\u53cc\u8bed\u80fd\u529b\uff08\u4e2d\u6587\/\u82f1\u6587\uff09\u800c\u8457\u79f0\uff0c\u7ecf\u8fc7\u5927\u89c4\u6a21\u7684\u591a\u8bed\u8a00\u8bed\u6599\u5e93\u8bad\u7ec3\uff0c\u5c55\u73b0\u51fa\u5353\u8d8a\u7684\u8bed\u8a00\u7406\u89e3\u548c\u63a8\u7406\u80fd\u529b\u3002Yi-6B \u6a21\u578b\u7279\u522b\u9002\u7528\u4e8e\u9700\u8981\u5904\u7406\u590d\u6742\u8bed\u8a00\u4efb\u52a1\u7684\u573a\u666f\uff0c\u5982\u6587\u672c\u751f\u6210\u3001\u8bed\u8a00\u7ffb\u8bd1\u548c\u77e5\u8bc6\u95ee\u7b54\u3002<\/p>\n<h3 id=\"-\">\u5173\u952e\u7279\u6027<\/h3>\n<ul>\n<li><strong>\u591a\u8bed\u8a00\u652f\u6301<\/strong>\uff1a\u7ecf\u8fc7\u591a\u8bed\u8a00\u8bed\u6599\u5e93\u8bad\u7ec3\uff0c\u80fd\u591f\u7406\u89e3\u548c\u751f\u6210\u4e2d\u6587\u548c\u82f1\u6587\u6587\u672c\u3002<\/li>\n<li><strong>\u9ad8\u6027\u80fd<\/strong>\uff1a\u5728\u591a\u4e2a\u57fa\u51c6\u6d4b\u8bd5\u4e2d\u8868\u73b0\u51fa\u8272\uff0c\u5305\u62ec\u8bed\u8a00\u7406\u89e3\u3001\u5e38\u8bc6\u63a8\u7406\u548c\u9605\u8bfb\u7406\u89e3\u3002<\/li>\n<li><strong>\u5f00\u6e90\u6027<\/strong>\uff1a\u4ee3\u7801\u548c\u6743\u91cd\u5728 Apache 2.0 \u8bb8\u53ef\u4e0b\u5206\u53d1\uff0c\u5141\u8bb8\u4e2a\u4eba\u3001\u5b66\u672f\u548c\u5546\u4e1a\u7528\u9014\u3002<\/li>\n<\/ul>\n<h2 id=\"-\">\u6a21\u578b\u8bc4\u6d4b<\/h2>\n<p>Yi-6B \u6a21\u578b\u5728\u591a\u4e2a\u57fa\u51c6\u6d4b\u8bd5\u4e2d\u8fdb\u884c\u4e86\u8bc4\u4f30\uff0c\u663e\u793a\u51fa\u5176\u5728\u4e0d\u540c\u4efb\u52a1\u4e0a\u7684\u5f3a\u5927\u6027\u80fd\u3002\u4ee5\u4e0b\u662f\u4e00\u4e9b\u5173\u952e\u7684\u8bc4\u6d4b\u7ed3\u679c\uff1a<\/p>\n<h3 id=\"-\">\u57fa\u51c6\u6d4b\u8bd5\u8868\u73b0<\/h3>\n<ul>\n<li><strong>\u8bed\u8a00\u7406\u89e3<\/strong>\uff1a\u5728\u8bed\u8a00\u7406\u89e3\u4efb\u52a1\u4e2d\uff0cYi-6B \u8868\u73b0\u51fa\u9ad8\u5ea6\u7684\u51c6\u786e\u6027\u548c\u7406\u89e3\u529b\u3002<\/li>\n<li><strong>\u5e38\u8bc6\u63a8\u7406<\/strong>\uff1a\u5728\u5e38\u8bc6\u63a8\u7406\u4efb\u52a1\u4e2d\uff0cYi-6B \u80fd\u591f\u51c6\u786e\u63a8\u65ad\u51fa\u903b\u8f91\u7ed3\u8bba\u3002<\/li>\n<li><strong>\u9605\u8bfb\u7406\u89e3<\/strong>\uff1a\u5728\u9605\u8bfb\u7406\u89e3\u4efb\u52a1\u4e2d\uff0cYi-6B \u80fd\u591f\u51c6\u786e\u56de\u7b54\u95ee\u9898\u5e76\u7406\u89e3\u6587\u672c\u7684\u6df1\u5c42\u542b\u4e49\u3002<\/li>\n<\/ul>\n<h3 id=\"-\">\u8bc4\u6d4b\u65b9\u6cd5<\/h3>\n<ul>\n<li><strong>\u96f6\u6837\u672c\u5b66\u4e60\uff08Zero-shot\uff09<\/strong>\uff1a\u5728\u6ca1\u6709\u8bad\u7ec3\u6570\u636e\u7684\u60c5\u51b5\u4e0b\uff0c\u76f4\u63a5\u5bf9\u6a21\u578b\u8fdb\u884c\u6d4b\u8bd5\u3002<\/li>\n<li><strong>\u5c11\u6837\u672c\u5b66\u4e60\uff08Few-shot\uff09<\/strong>\uff1a\u4f7f\u7528\u5c11\u91cf\u8bad\u7ec3\u6837\u672c\u5bf9\u6a21\u578b\u8fdb\u884c\u6d4b\u8bd5\u3002<\/li>\n<\/ul>\n<h2 id=\"-\">\u90e8\u7f72\u4f7f\u7528<\/h2>\n<p>Yi-6B \u6a21\u578b\u53ef\u4ee5\u901a\u8fc7\u591a\u79cd\u65b9\u5f0f\u8fdb\u884c\u90e8\u7f72\uff0c\u4ee5\u9002\u5e94\u4e0d\u540c\u7684\u4f7f\u7528\u573a\u666f\u3002\u4ee5\u4e0b\u662f\u4e00\u4e9b\u8be6\u7ec6\u7684\u90e8\u7f72\u6b65\u9aa4\uff1a<\/p>\n<h3 id=\"-pip-\">\u4f7f\u7528 pip \u90e8\u7f72<\/h3>\n<ol>\n<li><strong>\u5b89\u88c5 Python<\/strong>\uff1a\u786e\u4fdd\u7cfb\u7edf\u4e0a\u5b89\u88c5\u4e86 Python 3.10 \u6216\u66f4\u9ad8\u7248\u672c\u3002<\/li>\n<li><strong>\u51c6\u5907\u73af\u5883<\/strong>\uff1a\u4f7f\u7528\u4ee5\u4e0b\u547d\u4ee4\u8bbe\u7f6e\u73af\u5883\u5e76\u5b89\u88c5\u6240\u9700\u5305\uff1a\n<pre><code class=\"lang-bash\">pip <span class=\"token function\">install<\/span> transformers\r\n<\/code><\/pre>\n<\/li>\n<li><strong>\u4e0b\u8f7d\u6a21\u578b<\/strong>\uff1a\u4ece Hugging Face\u3001ModelScope \u6216 WiseModel \u4e0b\u8f7d Yi-6B \u6a21\u578b\u7684\u6743\u91cd\u548c\u5206\u8bcd\u5668\u3002<\/li>\n<li><strong>\u6267\u884c\u63a8\u7406<\/strong>\uff1a\u4f7f\u7528\u4ee5\u4e0b Python \u4ee3\u7801\u8fdb\u884c\u63a8\u7406\uff1a\n<pre><code class=\"lang-python\"><span class=\"token keyword\">from<\/span> transformers <span class=\"token keyword\">import<\/span> YiTokenizer<span class=\"token punctuation\">,<\/span> YiForCausalLM\r\n\r\ntokenizer <span class=\"token operator\">=<\/span> YiTokenizer<span class=\"token punctuation\">.<\/span>from_pretrained<span class=\"token punctuation\">(<\/span><span class=\"token string\">'model_path'<\/span><span class=\"token punctuation\">)<\/span>\r\nmodel <span class=\"token operator\">=<\/span> YiForCausalLM<span class=\"token punctuation\">.<\/span>from_pretrained<span class=\"token punctuation\">(<\/span><span class=\"token string\">'model_path'<\/span><span class=\"token punctuation\">)<\/span>\r\n\r\ninput_text <span class=\"token operator\">=<\/span> <span class=\"token string\">\"Hello, how are you?\"<\/span>\r\ntokens <span class=\"token operator\">=<\/span> tokenizer<span class=\"token punctuation\">.<\/span>encode<span class=\"token punctuation\">(<\/span>input_text<span class=\"token punctuation\">,<\/span> return_tensors<span class=\"token operator\">=<\/span><span class=\"token string\">\"pt\"<\/span><span class=\"token punctuation\">)<\/span>\r\noutput <span class=\"token operator\">=<\/span> model<span class=\"token punctuation\">.<\/span>generate<span class=\"token punctuation\">(<\/span>tokens<span class=\"token punctuation\">)<\/span>\r\nresponse <span class=\"token operator\">=<\/span> tokenizer<span class=\"token punctuation\">.<\/span>decode<span class=\"token punctuation\">(<\/span>output<span class=\"token punctuation\">[<\/span><span class=\"token number\">0<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span> skip_special_tokens<span class=\"token operator\">=<\/span><span class=\"token boolean\">True<\/span><span class=\"token punctuation\">)<\/span>\r\n<span class=\"token keyword\">print<\/span><span class=\"token punctuation\">(<\/span>response<span class=\"token punctuation\">)<\/span>\r\n<\/code><\/pre>\n<\/li>\n<\/ol>\n<h3 id=\"-docker-\">\u4f7f\u7528 Docker \u90e8\u7f72<\/h3>\n<ol>\n<li><strong>\u5b89\u88c5 Docker<\/strong>\uff1a\u786e\u4fdd\u7cfb\u7edf\u4e0a\u5b89\u88c5\u4e86 Docker \u548c nvidia-container-toolkit\u3002<\/li>\n<li><strong>\u542f\u52a8 Docker<\/strong>\uff1a\n<pre><code class=\"lang-bash\">docker run -it --gpus all \\\r\n-v <span class=\"token operator\">&lt;<\/span>your-model-path<span class=\"token operator\">&gt;<\/span>: \/models \\\r\nghcr.io\/01-ai\/yi:latest\r\n<\/code><\/pre>\n<\/li>\n<li><strong>\u6267\u884c\u63a8\u7406<\/strong>\uff1a\u5728 Docker \u5bb9\u5668\u5185\u6267\u884c\u63a8\u7406\u4ee3\u7801\u3002<\/li>\n<\/ol>\n<h3 id=\"-llama-cpp-\">\u4f7f\u7528 llama.cpp \u90e8\u7f72<\/h3>\n<ol>\n<li><strong>\u4e0b\u8f7d llama.cpp<\/strong>\uff1a\u4ece GitHub \u514b\u9686\u4ed3\u5e93\u3002<\/li>\n<li><strong>\u4e0b\u8f7d\u6a21\u578b<\/strong>\uff1a\u4e0b\u8f7d Yi-6B \u6a21\u578b\u7684\u91cf\u5316\u7248\u672c\u3002<\/li>\n<li><strong>\u6267\u884c\u63a8\u7406<\/strong>\uff1a\u4f7f\u7528\u547d\u4ee4\u884c\u6216 Web \u754c\u9762\u8fdb\u884c\u63a8\u7406\u3002<\/li>\n<\/ol>\n<h2 id=\"-\">\u5e38\u89c1\u95ee\u9898<\/h2>\n<h3 id=\"-\">\u5982\u4f55\u8fdb\u884c\u6a21\u578b\u5fae\u8c03\uff1f<\/h3>\n<ul>\n<li><strong>\u51c6\u5907\u6570\u636e\u96c6<\/strong>\uff1a\u51c6\u5907\u4e00\u4e2a\u5305\u542b\u8bad\u7ec3\u793a\u4f8b\u7684 JSONL \u683c\u5f0f\u6570\u636e\u96c6\u3002<\/li>\n<li><strong>\u8bbe\u7f6e\u73af\u5883<\/strong>\uff1a\u4f7f\u7528 conda \u521b\u5efa\u4e00\u4e2a\u73af\u5883\u5e76\u5b89\u88c5\u4f9d\u8d56\u3002<\/li>\n<li><strong>\u8bad\u7ec3\u6a21\u578b<\/strong>\uff1a\u4f7f\u7528\u63d0\u4f9b\u7684\u811a\u672c\u8fdb\u884c\u5fae\u8c03\u8bad\u7ec3\u3002<\/li>\n<\/ul>\n<h3 id=\"-\">\u5982\u4f55\u91cf\u5316\u6a21\u578b\uff1f<\/h3>\n<ul>\n<li><strong>\u4f7f\u7528 GPT-Q<\/strong>\uff1a\u4f7f\u7528 AutoGPTQ \u548c exllama \u8fdb\u884c GPT-Q \u91cf\u5316\u3002<\/li>\n<li><strong>\u4f7f\u7528 AWQ<\/strong>\uff1a\u4f7f\u7528 AutoAWQ \u8fdb\u884c AWQ \u91cf\u5316\u3002<\/li>\n<\/ul>\n<h2 id=\"-\">\u76f8\u5173\u8d44\u6e90<\/h2>\n<h3 id=\"-\">\u8bbf\u95ee\u94fe\u63a5<\/h3>\n<ul>\n<li><strong>Hugging Face<\/strong>:\u00a0<a target=\"_blank\" rel=\"noopener noreferrer nofollow\" href=\"https:\/\/huggingface.co\/01-ai\/Yi-6B\">Yi-6B on Hugging Face<\/a><\/li>\n<li><strong>ModelScope<\/strong>:\u00a0<a target=\"_blank\" rel=\"noopener noreferrer nofollow\" href=\"https:\/\/modelscope.cn\/models\/01ai\/Yi-6B\">Yi-6B on ModelScope<\/a><\/li>\n<li><strong>WiseModel<\/strong>:\u00a0<a target=\"_blank\" rel=\"noopener noreferrer nofollow\" href=\"https:\/\/wisemodel.cn\/01ai\/Yi-6B\">Yi-6B on WiseModel<\/a><\/li>\n<\/ul>\n<h3 id=\"-\">\u793e\u533a\u548c\u652f\u6301<\/h3>\n<ul>\n<li><strong>GitHub<\/strong>:\u00a0<a target=\"_blank\" rel=\"noopener noreferrer nofollow\" href=\"https:\/\/github.com\/01-ai\/Yi\">Yi Model GitHub<\/a><\/li>\n<li><strong>Discord<\/strong>:\u00a0<a target=\"_blank\" rel=\"noopener noreferrer nofollow\" href=\"https:\/\/discord.gg\/01ai\">Yi Model Discord<\/a><\/li>\n<li><strong>WeChat<\/strong>:\u00a0<a target=\"_blank\" rel=\"noopener noreferrer nofollow\" href=\"https:\/\/weixin.qq.com\/01ai\">Yi Model WeChat Group<\/a><\/li>\n<\/ul>\n<p>\u901a\u8fc7\u8fd9\u4e9b\u8d44\u6e90\uff0c\u53ef\u4ee5\u83b7\u53d6\u66f4\u591a\u5173\u4e8e Yi-6B \u6a21\u578b\u7684\u4fe1\u606f\u548c\u652f\u6301\u3002<\/p>\n","protected":false},"excerpt":{"rendered":"<p>Yi-6B \u662f\u7531 01.AI \u96f6\u4e00\u4e07\u7269\u5f00\u53d1\u7684\u4e00\u7cfb\u5217\u5927\u578b\u8bed\u8a00\u6a21\u578b\uff08LLMs\uff09\u4e4b\u4e00\u3002\u8fd9\u4e00\u7cfb\u5217\u6a21\u578b\u4ee5\u5176\u5f00\u6e90\u6027\u548c\u53cc\u8bed\u80fd\u529b\uff08\u4e2d\u6587\/\u82f1\u6587\uff09\u800c\u8457\u79f0\uff0c\u7ecf\u8fc7\u5927\u89c4\u6a21\u7684\u591a\u8bed\u8a00\u8bed\u6599\u5e93\u8bad\u7ec3\uff0c\u5c55\u73b0\u51fa\u5353\u8d8a\u7684\u8bed\u8a00\u7406\u89e3\u548c\u63a8\u7406\u80fd\u529b\u3002Yi-6B \u6a21\u578b\u7279\u522b\u9002\u7528\u4e8e\u9700\u8981\u5904\u7406\u590d\u6742\u8bed\u8a00\u4efb\u52a1\u7684\u573a\u666f\uff0c\u5982\u6587\u672c\u751f\u6210\u3001\u8bed\u8a00\u7ffb\u8bd1\u548c\u77e5\u8bc6\u95ee\u7b54\u3002 \u5173\u952e\u7279\u6027 \u591a\u8bed\u8a00\u652f\u6301\uff1a\u7ecf\u8fc7\u591a\u8bed\u8a00\u8bed\u6599\u5e93\u8bad\u7ec3\uff0c\u80fd\u591f\u7406\u89e3\u548c\u751f\u6210\u4e2d\u6587\u548c\u82f1\u6587\u6587\u672c\u3002 \u9ad8\u6027\u80fd\uff1a\u5728\u591a\u4e2a\u57fa\u51c6\u6d4b\u8bd5\u4e2d\u8868\u73b0\u51fa\u8272\uff0c\u5305\u62ec\u8bed\u8a00\u7406\u89e3\u3001\u5e38\u8bc6\u63a8\u7406\u548c\u9605\u8bfb\u7406\u89e3\u3002 \u5f00\u6e90\u6027\uff1a\u4ee3\u7801\u548c\u6743\u91cd\u5728 Apache 2.0 \u8bb8\u53ef\u4e0b\u5206\u53d1\uff0c\u5141\u8bb8\u4e2a\u4eba\u3001\u5b66\u672f\u548c\u5546\u4e1a\u7528\u9014\u3002 \u6a21\u578b\u8bc4<\/p>\n","protected":false},"author":1,"featured_media":121,"comment_status":"open","ping_status":"closed","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[3],"tags":[],"collection":[],"company":[22],"rank":[],"class_list":["post-120","post","type-post","status-publish","format-standard","has-post-thumbnail","hentry","category-llm","company-01-ai"],"_links":{"self":[{"href":"https:\/\/www.onetts.com\/ai\/wp-json\/wp\/v2\/posts\/120","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/www.onetts.com\/ai\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/www.onetts.com\/ai\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/www.onetts.com\/ai\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/www.onetts.com\/ai\/wp-json\/wp\/v2\/comments?post=120"}],"version-history":[{"count":1,"href":"https:\/\/www.onetts.com\/ai\/wp-json\/wp\/v2\/posts\/120\/revisions"}],"predecessor-version":[{"id":122,"href":"https:\/\/www.onetts.com\/ai\/wp-json\/wp\/v2\/posts\/120\/revisions\/122"}],"wp:featuredmedia":[{"embeddable":true,"href":"https:\/\/www.onetts.com\/ai\/wp-json\/wp\/v2\/media\/121"}],"wp:attachment":[{"href":"https:\/\/www.onetts.com\/ai\/wp-json\/wp\/v2\/media?parent=120"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/www.onetts.com\/ai\/wp-json\/wp\/v2\/categories?post=120"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/www.onetts.com\/ai\/wp-json\/wp\/v2\/tags?post=120"},{"taxonomy":"collection","embeddable":true,"href":"https:\/\/www.onetts.com\/ai\/wp-json\/wp\/v2\/collection?post=120"},{"taxonomy":"company","embeddable":true,"href":"https:\/\/www.onetts.com\/ai\/wp-json\/wp\/v2\/company?post=120"},{"taxonomy":"rank","embeddable":true,"href":"https:\/\/www.onetts.com\/ai\/wp-json\/wp\/v2\/rank?post=120"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}