{"id":1393,"date":"2024-05-09T03:32:04","date_gmt":"2024-05-09T03:32:04","guid":{"rendered":"https:\/\/www.nicekj.com\/?p=1393"},"modified":"2024-05-09T03:32:23","modified_gmt":"2024-05-09T03:32:23","slug":"tongguo-4-bit-lianghuajiazaiheyunxing-mistral-7b-ai","status":"publish","type":"post","link":"https:\/\/www.nicekj.com\/tongguo-4-bit-lianghuajiazaiheyunxing-mistral-7b-ai.html","title":{"rendered":"\u901a\u8fc7 4-bit \u91cf\u5316\u52a0\u8f7d\u548c\u8fd0\u884c Mistral 7B AI"},"content":{"rendered":"<p>\n<figure class=\"wp-block-image size-large\"><img decoding=\"async\" src=\"https:\/\/www.nicekj.com\/wp-content\/uploads\/replace\/535ad15d3cfb483e22dbe07b05b9daaf.png\" alt=\"\u672a\u6807\u9898-2.jpg\" \/><\/figure>\n<\/p>\n<p>\u5927\u8bed\u8a00\u6a21\u578b (LLM) \u7684\u4e16\u754c\u6b63\u5728\u5feb\u901f\u53d1\u5c55\uff0c\u50cf LLAMA2 \u548c Falcon \u8fd9\u6837\u7684\u7a81\u7834\u6027\u6a21\u578b\u4e0d\u65ad\u51fa\u73b0\u3002<\/p>\n<p>\u53bb\u5e74 9 \u6708 27 \u65e5\uff0c\u6cd5\u56fd\u521d\u521b\u516c\u53f8 Mistral \u63a8\u51fa\u4e86\u7b2c\u4e00\u4e2a\u6a21\u578b \u2014\u2014 <code>Mistral 7B<\/code> \uff0c\u5f15\u8d77\u4e86\u79d1\u6280\u754c\u7684\u8f70\u52a8\uff0c\u5e76\u83b7\u5f97\u4e86\u8fc4\u4eca\u4e3a\u6b62\u540c\u7c7b\u89c4\u6a21\u4e2d\u6700\u5f3a\u5927\u7684\u8bed\u8a00\u6a21\u578b\u7684\u79f0\u53f7\u3002<\/p>\n<p>\u672c\u6587\u5c06\u5c55\u793a\u901a\u8fc7 <code>4-bit<\/code> \u91cf\u5316\u52a0\u8f7d\u548c\u8fd0\u884c\u8fd0\u884c <code>Mistral 7B AI<\/code><\/p>\n<h3 data-id=\"heading-0\">\u4ec0\u4e48\u662fMistral 7B\uff1f<\/h3>\n<p><a href=\"https:\/\/link.juejin.cn?target=https%3A%2F%2Fmistral.ai%2Fnews%2Fannouncing-mistral-7b%2F\" target=\"_blank\" title=\"https:\/\/mistral.ai\/news\/announcing-mistral-7b\/\" ref=\"nofollow noopener noreferrer\" rel=\"noopener\">Mistral-7B-v0.1<\/a> \u662f\u5bf9 <code>Llama 2<\/code> \u7b49\u5176\u4ed6 <code>\u5c0f\u578b<\/code> \u5927\u578b\u8bed\u8a00\u6a21\u578b\u7684\u8fdb\u4e00\u6b65\u6539\u8fdb\uff0c\u4ee5\u76f8\u5f53\u4f4e\u7684\u8ba1\u7b97\u6210\u672c\u63d0\u4f9b\u7c7b\u4f3c\u7684\u529f\u80fd\uff08\u6839\u636e\u4e00\u4e9b\u6807\u51c6\u57fa\u51c6\uff09\u3002\u53ef\u9002\u5e94\u8bb8\u591a\u573a\u666f\uff0c\u53ef\u4ee5\u6267\u884c\u5404\u79cd\u81ea\u7136\u8bed\u8a00\u5904\u7406\u4efb\u52a1\uff0c\u5e8f\u5217\u957f\u5ea6\u4e3a <code>8k<\/code>\u3002\u4f8b\u5982\uff0c\u5b83\u975e\u5e38\u9002\u5408\u6587\u672c\u6458\u8981\u3001\u5206\u7c7b\u3001\u6587\u672c\u8865\u5168\u3001\u4ee3\u7801\u8865\u5168\u3002<\/p>\n<p>\u4e0b\u9762\u662f Mistral 7B \u7684\u7279\u5f81\uff1a<\/p>\n<ul>\n<li>\n<p><strong>\u65e0\u4e0e\u4f26\u6bd4\u7684\u6027\u80fd<\/strong>\uff1a<code>Mistral 7B<\/code> \u5728\u6240\u6709\u57fa\u51c6\u6d4b\u8bd5\u4e2d\u5747\u4f18\u4e8e <code>Llama 2 13B<\/code>\u3002<\/p>\n<p>\n<figure class=\"wp-block-image size-large\"><img decoding=\"async\" src=\"https:\/\/www.nicekj.com\/wp-content\/uploads\/replace\/ee211378c26fda7d9d3bb2dfa660e30b.png\" alt=\"image.png\" \/><\/figure>\n<\/p>\n<\/li>\n<li>\n<p><strong>\u66f4\u9ad8\u6548<\/strong>\uff1a\u5f97\u76ca\u4e8e\u5206\u7ec4\u67e5\u8be2\u6ce8\u610f\u529b (GQA) \u548c\u6ed1\u52a8\u7a97\u53e3\u6ce8\u610f\u529b (SWA)\uff0c<code>Mistral 7B<\/code> \u53ef\u4ee5\u63d0\u4f9b\u66f4\u5feb\u7684\u63a8\u7406\u901f\u5ea6\u5e76\u8f7b\u677e\u5904\u7406\u66f4\u957f\u7684\u5e8f\u5217\u3002<\/p>\n<\/li>\n<li>\n<p><strong>\u5f00\u6e90<\/strong>\uff1a<code>Mistral 7B<\/code> \u5728 Apache 2.0 \u8bb8\u53ef\u8bc1\u4e0b\u53d1\u5e03\uff0c\u53ef\u4ee5\u4e0d\u53d7\u9650\u5236\u5730\u4f7f\u7528\u3002<\/p>\n<\/li>\n<\/ul>\n<h3 data-id=\"heading-1\">\u4ec0\u4e48\u662f\u91cf\u5316 quantization \u548c QLoRA\uff1f<\/h3>\n<p>\u4e0e\u540c\u7c7b\u4ea7\u54c1\u76f8\u6bd4\uff0c<code>Mistral 7B<\/code> \u53ef\u80fd\u66f4\u5c0f\uff0c\u4f46\u8ba9\u5b83\u5728\u6d88\u8d39\u7c7b\u786c\u4ef6\u4e2d\u8fd0\u884c\u548c\u8bad\u7ec3\u4ecd\u7136\u662f\u4e00\u4e2a\u5de8\u5927\u7684\u6311\u6218\u3002<\/p>\n<p>\u4e3a\u4e86\u80fd\u5728\u5355\u4e2a GPU \u4e0a\u8fd0\u884c\u5b83\uff0c\u9700\u8981\u4ee5 <code>4-bit<\/code> \u7cbe\u5ea6\u8fd0\u884c\u6a21\u578b\uff0c\u5e76\u4f7f\u7528 QLoRA \u6765\u51cf\u5c11\u5185\u5b58\u4f7f\u7528\u3002<\/p>\n<h4 data-id=\"heading-2\">QLoRA \u89e3\u51b3\u65b9\u6848<\/h4>\n<p>QLoRA \u4ee3\u8868\u5177\u6709\u4f4e\u9636\u9002\u914d\u5668\u7684\u91cf\u5316 LLM\uff0c\u662f\u4e00\u79cd\u6709\u6548\u7684\u5fae\u8c03\u65b9\u6cd5\u3002\u5b83\u4f7f\u7528 <code>4-bit<\/code> \u91cf\u5316\u6765\u538b\u7f29\u9884\u8bad\u7ec3\u7684\u8bed\u8a00\u6a21\u578b\uff0c\u4e0e\u6807\u51c6 <code>16-bit<\/code> \u6a21\u578b\u5fae\u8c03\u76f8\u6bd4\uff0c\u65e0\u9700\u727a\u7272\u6027\u80fd\u3002<\/p>\n<blockquote>\n<p><strong>QLoRA\u8bba\u6587\u6458\u8981\uff1a<\/strong> \u6211\u4eec\u63d0\u51fa\u4e86 QLoRA\uff0c\u8fd9\u662f\u4e00\u79cd\u9ad8\u6548\u7684\u5fae\u8c03\u65b9\u6cd5\uff0c\u53ef\u51cf\u5c11\u5185\u5b58\u4f7f\u7528\u91cf\uff0c\u8db3\u4ee5\u5728\u5355\u4e2a <code>48GB GPU<\/code> \u4e0a\u5fae\u8c03 <code>65B<\/code> \u53c2\u6570\u7684\u6a21\u578b\uff0c\u540c\u65f6\u4fdd\u7559\u5b8c\u6574\u7684 <code>16-bit<\/code> \u5fae\u8c03\u4efb\u52a1\u6027\u80fd\u3002QLoRA \u5f15\u5165\u4e86\u591a\u9879\u521b\u65b0\u6280\u672f\u6765\u8282\u7701\u5185\u5b58\u800c\u4e0d\u727a\u7272\u6027\u80fd\uff1aa\uff09<code>4-bit<\/code> \u666e\u901a\u6d6e\u70b9\u578b <code>NF4<\/code> \u4e00\u79cd\u65b0\u7684\u6570\u636e\u7c7b\u578b\uff0c\u7406\u8bba\u4e0a\u5bf9\u4e8e\u6b63\u6001\u5206\u5e03\u6743\u91cd\u6765\u8bf4\u662f\u6700\u4f73\u4fe1\u606f\uff1bb\uff09\u53cc\u91cf\u5316\u901a\u8fc7\u91cf\u5316\u91cf\u5316\u5e38\u6570\u6765\u51cf\u5c11\u5e73\u5747\u5185\u5b58\u5360\u7528\uff0c\u4ee5\u53ca\u5206\u9875\u4f18\u5316\u5668\u6765\u7ba1\u7406\u5185\u5b58\u5cf0\u503c\u3002<\/p>\n<\/blockquote>\n<h3 data-id=\"heading-3\">\u4f7f\u7528\u6b65\u9aa4<\/h3>\n<h4 data-id=\"heading-4\">\u7b2c 1 \u6b65\uff1a\u5b89\u88c5\u5fc5\u8981\u7684\u8f6f\u4ef6\u5305<\/h4>\n<p>QLoRA \u4f7f\u7528 <a href=\"https:\/\/link.juejin.cn?target=https%3A%2F%2Fgithub.com%2FTimDettmers%2Fbitsandbytes\" target=\"_blank\" title=\"https:\/\/github.com\/TimDettmers\/bitsandbytes\" ref=\"nofollow noopener noreferrer\" rel=\"noopener\">bitsandbytes<\/a> \u8fdb\u884c\u91cf\u5316\uff0c\u5e76\u4e0e Hugging Face \u7684 <a href=\"https:\/\/link.juejin.cn?target=https%3A%2F%2Fgithub.com%2Fhuggingface%2Fpeft\" target=\"_blank\" title=\"https:\/\/github.com\/huggingface\/peft\" ref=\"nofollow noopener noreferrer\" rel=\"noopener\">PEFT<\/a> \u548c<a href=\"https:\/\/link.juejin.cn?target=https%3A%2F%2Fgithub.com%2Fhuggingface%2Ftransformers%2F\" target=\"_blank\" title=\"https:\/\/github.com\/huggingface\/transformers\/\" ref=\"nofollow noopener noreferrer\" rel=\"noopener\">transformers<\/a> \u5e93\u96c6\u6210\u3002<\/p>\n<p>\u8fd9\u91cc\u5e0c\u671b\u786e\u4fdd\u4f7f\u7528\u6700\u65b0\u529f\u80fd\uff0c\u5c06\u4ece\u6e90\u4ee3\u7801\u5b89\u88c5\u8fd9\u4e9b\u4f9d\u8d56\u5e93\u3002<\/p>\n<pre><\/div><div class=\"code-block-extension-headerRight\"><span class=\"code-block-extension-lang\">ruby<\/span><div class=\"code-block-extension-copyCodeBtn\">\u590d\u5236\u4ee3\u7801<\/div><\/div><\/div><code class=\"hljs language-ruby code-block-extension-codeShowNum\" lang=\"ruby\"><span class=\"code-block-extension-codeLine\" data-line-num=\"1\">pip install -q -U bitsandbytes<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"2\">pip install -q -U git+<span class=\"hljs-symbol\">https:<\/span>\/<span class=\"hljs-regexp\">\/github.com\/huggingface<\/span><span class=\"hljs-regexp\">\/transformers.git<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"3\">pip install -q -U git+https:\/<span class=\"hljs-regexp\">\/github.com\/huggingface<\/span><span class=\"hljs-regexp\">\/peft.git<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"4\">pip install -q -U git+https:\/<span class=\"hljs-regexp\">\/github.com\/huggingface<\/span><span class=\"hljs-regexp\">\/accelerate.git<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"5\"><\/span>\n<\/code><\/pre>\n<h4 data-id=\"heading-5\">\u7b2c 2 \u6b65\uff1a\u901a\u8fc7 Transformer \u7684 BitsandBytesConfig \u5b9a\u4e49\u91cf\u5316\u53c2\u6570<\/h4>\n<p>\u73b0\u5728\u5c06\u4f7f\u7528 Transformer \u5e93\u4e2d\u7684 BitsandBytesConfig \u914d\u7f6e QLoRA \u53c2\u6570\u3002<\/p>\n<p>\u8fd9\u91cc\u662f\u5bf9\u53ef\u4ee5\u8c03\u6574\u548c\u4f7f\u7528\u7684\u53c2\u6570\u8fdb\u884c\u8bf4\u660e\uff1a<\/p>\n<ul>\n<li><code>load_in_4bit=True<\/code>\uff1a\u6307\u5b9a\u8981\u4ee5 <code>4-bit<\/code> \u7cbe\u5ea6\u8f6c\u6362\u548c\u52a0\u8f7d\u6a21\u578b\u3002<\/li>\n<li><code>bnb_4bit_use_double_quant=True<\/code>\uff1a\u4f7f\u7528\u5d4c\u5957\u91cf\u5316\u6765\u63d0\u9ad8\u5185\u5b58\u6548\u7387\u7684\u63a8\u7406\u548c\u8bad\u7ec3\u3002<\/li>\n<li><code>bnd_4bit_quant_type=\"nf4\"<\/code>\uff1a<code>4-bit<\/code> \u96c6\u6210\u5e26\u6709 2 \u79cd\u4e0d\u540c\u7684\u91cf\u5316\u7c7b\u578b <code>FP4<\/code> \u548c <code>NF4<\/code>\u3002 <code>NF4 dtype<\/code> \u4ee3\u8868 <code>Normal Float 4<\/code>\uff0c\u5728 <a href=\"https:\/\/link.juejin.cn?target=https%3A%2F%2Farxiv.org%2Fabs%2F2305.14314\" target=\"_blank\" title=\"https:\/\/arxiv.org\/abs\/2305.14314\" ref=\"nofollow noopener noreferrer\" rel=\"noopener\">QLoRA \u8bba\u6587<\/a>\u4e2d\u6709\u4ecb\u7ecd\u3002\u9ed8\u8ba4\u60c5\u51b5\u4e0b\uff0c\u4f7f\u7528 <code>FP4<\/code> \u91cf\u5316\u3002<\/li>\n<li><code>bnb_4bit_compute_dype=torch.bfloat16<\/code>\uff1a\u8ba1\u7b97\u6570\u636e\u7c7b\u578b\u7528\u4e8e\u66f4\u6539\u8ba1\u7b97\u671f\u95f4\u5c06\u4f7f\u7528\u7684\u6570\u636e\u7c7b\u578b\u3002\u9ed8\u8ba4\u60c5\u51b5\u4e0b\uff0c\u8ba1\u7b97\u6570\u636e\u7c7b\u578b\u8bbe\u7f6e\u4e3a <code>float32<\/code>\uff0c\u4f46\u53ef\u4ee5\u8bbe\u7f6e\u4e3a <code>bf16<\/code> \u4ee5\u63d0\u9ad8\u901f\u5ea6\u3002<\/li>\n<\/ul>\n\n<pre><\/div><div class=\"code-block-extension-headerRight\"><span class=\"code-block-extension-lang\">ini<\/span><div class=\"code-block-extension-copyCodeBtn\">\u590d\u5236\u4ee3\u7801<\/div><\/div><\/div><code class=\"hljs language-ini code-block-extension-codeShowNum\" lang=\"ini\"><span class=\"code-block-extension-codeLine\" data-line-num=\"1\">import torch<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"2\">from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"3\"><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"4\"><span class=\"hljs-attr\">bnb_config<\/span> = BitsAndBytesConfig(<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"5\">    <span class=\"hljs-attr\">load_in_4bit<\/span>=<span class=\"hljs-literal\">True<\/span>,<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"6\">    <span class=\"hljs-attr\">bnb_4bit_use_double_quant<\/span>=<span class=\"hljs-literal\">True<\/span>,<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"7\">    <span class=\"hljs-attr\">bnb_4bit_quant_type<\/span>=<span class=\"hljs-string\">\"nf4\"<\/span>,<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"8\">    <span class=\"hljs-attr\">bnb_4bit_compute_dtype<\/span>=torch.bfloat16<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"9\">)<\/span>\n<\/code><\/pre>\n<h4 data-id=\"heading-6\">\u7b2c 3 \u6b65\uff1a\u4e3a Mistral 7B \u52a0\u8f7d\u91cf\u5316<\/h4>\n<p>\u73b0\u5728\u6765\u6307\u5b9a\u6a21\u578b ID\uff0c\u7136\u540e\u4f7f\u7528\u4e4b\u524d\u5b9a\u4e49\u7684\u91cf\u5316\u914d\u7f6e\u52a0\u8f7d\u5b83\u3002<\/p>\n<pre><\/div><div class=\"code-block-extension-headerRight\"><span class=\"code-block-extension-lang\">ini<\/span><div class=\"code-block-extension-copyCodeBtn\">\u590d\u5236\u4ee3\u7801<\/div><\/div><\/div><code class=\"hljs language-ini code-block-extension-codeShowNum\" lang=\"ini\"><span class=\"code-block-extension-codeLine\" data-line-num=\"1\"><span class=\"hljs-attr\">model_id<\/span> = <span class=\"hljs-string\">\"mistralai\/Mistral-7B-Instruct-v0.1\"<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"2\"><span class=\"hljs-attr\">model<\/span> = AutoModelForCausalLM.from_pretrained(model_id, quantization_config=bnb_config, device_map=<span class=\"hljs-string\">\"auto\"<\/span>)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"3\"><span class=\"hljs-attr\">tokenizer<\/span> = AutoTokenizer.from_pretrained(model_id)<\/span>\n<\/code><\/pre>\n<h4 data-id=\"heading-7\">\u7b2c 4 \u6b65\uff1a\u52a0\u8f7d\u540e\uff0c\u8fd0\u884c\u4e00\u4ee3\u5e76\u5c1d\u8bd5\u4e00\u4e0b\uff01<\/h4>\n<p>\u6700\u540e\uff0c\u51c6\u5907\u597d\u5c06 Mistral 7B \u6295\u5165\u4f7f\u7528\u3002<\/p>\n<ol>\n<li>\u9996\u5148\u6d4b\u8bd5\u5b83\u7684\u6587\u672c\u751f\u6210\u80fd\u529b\uff0c\u4f7f\u7528\u4ee5\u4e0b\u6a21\u677f\uff1a<\/li>\n<\/ol>\n\n<pre><\/div><div class=\"code-block-extension-headerRight\"><span class=\"code-block-extension-lang\">ini<\/span><div class=\"code-block-extension-copyCodeBtn\">\u590d\u5236\u4ee3\u7801<\/div><\/div><\/div><code class=\"hljs language-ini code-block-extension-codeShowNum\" lang=\"ini\"><span class=\"code-block-extension-codeLine\" data-line-num=\"1\"><span class=\"hljs-attr\">PROMPT<\/span>= <span class=\"hljs-string\">\"\"\" ### Instruction: Act as a data science expert.<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"2\">### Question:<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"3\">Explain to me what is Large Language Model. Assume that I am a 5-year-old child.<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"4\"><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"5\">### Answer:<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"6\">\"\"\"<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"7\"><span class=\"hljs-attr\">device<\/span> = <span class=\"hljs-string\">\"cuda:0\"<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"8\"><span class=\"hljs-attr\">encodeds<\/span> = tokenizer(PROMPT, return_tensors=<span class=\"hljs-string\">\"pt\"<\/span>, add_special_tokens=<span class=\"hljs-literal\">True<\/span>)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"9\"><span class=\"hljs-attr\">model_inputs<\/span> = encodeds.to(device)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"10\"><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"11\"><span class=\"hljs-attr\">generated_ids<\/span> = model.generate(**model_inputs, max_new_tokens=<span class=\"hljs-number\">1000<\/span>, do_sample=<span class=\"hljs-literal\">True<\/span>, pad_token_id=tokenizer.eos_token_id)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"12\"><span class=\"hljs-attr\">decoded<\/span> = tokenizer.batch_decode(generated_ids)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"13\">print(decoded<span class=\"hljs-section\">[0]<\/span>)<\/span>\n<\/code><\/pre>\n<p>\u8be5\u6a21\u578b\u9075\u5faa\u4e86\u6211\u4eec\u7684\u6307\u793a\uff0c\u5e76\u5f88\u597d\u5730\u89e3\u91ca\u4e86\u5927\u8bed\u8a00\u6a21\u578b\u7684\u6982\u5ff5\uff01<\/p>\n<p>\n<figure class=\"wp-block-image size-large\"><img decoding=\"async\" src=\"https:\/\/www.nicekj.com\/wp-content\/uploads\/replace\/3f99d29ea6bd13cdb5052f2c52862416.png\" alt=\"image.png\" \/><\/figure>\n<\/p>\n<ol start=\"2\">\n<li>\u73b0\u5728\u6765\u6d4b\u8bd5\u4e00\u4e0b Mistral 7B \u7684\u4ee3\u7801\u80fd\u529b<\/li>\n<\/ol>\n\n<pre><\/div><div class=\"code-block-extension-headerRight\"><span class=\"code-block-extension-lang\">ini<\/span><div class=\"code-block-extension-copyCodeBtn\">\u590d\u5236\u4ee3\u7801<\/div><\/div><\/div><code class=\"hljs language-ini code-block-extension-codeShowNum\" lang=\"ini\"><span class=\"code-block-extension-codeLine\" data-line-num=\"1\"><span class=\"hljs-attr\">messages<\/span> = [<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"2\">    {<span class=\"hljs-string\">\"role\"<\/span>: <span class=\"hljs-string\">\"user\"<\/span>, <span class=\"hljs-string\">\"content\"<\/span>: <span class=\"hljs-string\">\"write a python function to generate a list of random 1000 numbers between 1 and 10000?\"<\/span>}<span class=\"hljs-string\">\"role\"<\/span>: <span class=\"hljs-string\">\"user\"<\/span>, <span class=\"hljs-string\">\"content\"<\/span>: <span class=\"hljs-string\">\"write a python function to generate a list of random 1000 numbers between 1 and 10000?\"<\/span>}<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"3\">    ]<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"4\"><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"5\"><span class=\"hljs-attr\">encodeds<\/span> = tokenizer.apply_chat_template(messages, return_tensors=<span class=\"hljs-string\">\"pt\"<\/span>)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"6\"><span class=\"hljs-attr\">device<\/span> = <span class=\"hljs-string\">\"cuda:0\"<\/span><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"7\"><span class=\"hljs-attr\">model_inputs<\/span> = encodeds.to(device)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"8\"><\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"9\"><span class=\"hljs-attr\">generated_ids<\/span> = model.generate(model_inputs, max_new_tokens=<span class=\"hljs-number\">1000<\/span>, do_sample=<span class=\"hljs-literal\">True<\/span>)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"10\"><span class=\"hljs-attr\">decoded<\/span> = tokenizer.batch_decode(generated_ids)<\/span>\n<span class=\"code-block-extension-codeLine\" data-line-num=\"11\">print(decoded<span class=\"hljs-section\">[0]<\/span>)<\/span>\n<\/code><\/pre>\n<p>\u770b\u6765\u6a21\u578b\u5b8c\u6210\u7684\u4ee3\u7801\u6548\u679c\uff1a<\/p>\n<p>\n<figure class=\"wp-block-image size-large\"><img decoding=\"async\" src=\"https:\/\/www.nicekj.com\/wp-content\/uploads\/replace\/470aa67e4d592e3240e1a494a33ec498.png\" alt=\"image.png\" \/><\/figure>\n<\/p>\n<h3 data-id=\"heading-8\">\u603b\u7ed3<\/h3>\n<p>\u4ece\u76ee\u524d\u6a21\u578b\u7684\u53d1\u5c55\u60c5\u51b5\u6765\u770b\uff0cMistral AI \u662f LLaMA \u548c Falcon \u7b49\u6d41\u884c\u6a21\u578b\u7684\u4e00\u4e2a\u975e\u5e38\u4e0d\u9519\u7684\u66ff\u4ee3\u54c1\u3002<\/p>\n<p>\u5b83\u662f\u514d\u8d39\u7684\u3001\u66f4\u5c0f\u4f46\u66f4\u9ad8\u6548\u3002\u5b83\u5141\u8bb8\u5b8c\u5168\u5b9a\u5236\uff0c\u5e76\u4e14\u53ef\u4ee5\u8f7b\u677e\u5fae\u8c03\u5e76\u5177\u6709\u4ee4\u4eba\u60ca\u559c\u7684\u6027\u80fd\u3002<\/p>","protected":false},"excerpt":{"rendered":"<p>Mistral-7B-v0.1 \u662f\u5bf9 Llama 2 \u7b49\u5176\u4ed6 \u5c0f\u578b \u5927\u578b\u8bed\u8a00\u6a21\u578b\u7684\u8fdb\u4e00\u6b65\u6539\u8fdb\uff0c\u53ef\u9002\u5e94\u8bb8\u591a\u573a\u666f\uff0c\u53ef\u4ee5\u6267\u884c\u5404\u79cd\u81ea\u7136\u8bed\u8a00\u5904\u7406\u4efb\u52a1\uff0c\u4f8b\u5982\uff0c\u5b83\u975e\u5e38\u9002\u5408\u6587\u672c\u6458\u8981\u3001\u5206\u7c7b\u3001\u6587\u672c\u8865\u5168\u3001\u4ee3\u7801\u8865\u5168\u3002<\/p>\n","protected":false},"author":1,"featured_media":8240,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"_acf_changed":false,"rank_math_title":"","rank_math_description":"","rank_math_focus_keyword":"","views":"5","footnotes":""},"categories":[3],"tags":[126,127,128,129,136],"collection":[],"class_list":["post-1393","post","type-post","status-publish","format-standard","has-post-thumbnail","hentry","category-fenlei2","tag-gpt","tag-ai","tag-128","tag-129","tag-136"],"acf":[],"_links":{"self":[{"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/posts\/1393","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/comments?post=1393"}],"version-history":[{"count":0,"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/posts\/1393\/revisions"}],"wp:featuredmedia":[{"embeddable":true,"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/media\/8240"}],"wp:attachment":[{"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/media?parent=1393"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/categories?post=1393"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/tags?post=1393"},{"taxonomy":"collection","embeddable":true,"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/collection?post=1393"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}