{"id":1362,"date":"2023-11-25T23:02:25","date_gmt":"2023-11-25T15:02:25","guid":{"rendered":"https:\/\/www.kumouse.com\/?p=1362"},"modified":"2023-11-25T23:02:25","modified_gmt":"2023-11-25T15:02:25","slug":"%e7%99%be%e5%b7%9d2_baichuan2","status":"publish","type":"post","link":"https:\/\/www.kumouse.com\/?p=1362","title":{"rendered":"\u767e\u5ddd2_baichuan2"},"content":{"rendered":"<p>import torch<br \/>\nfrom modelscope import snapshot_download, AutoModelForCausalLM, AutoTokenizer,GenerationConfig<br \/>\nfrom transformers import BitsAndBytesConfig<\/p>\n<p>quantization_config = BitsAndBytesConfig(<br \/>\nFalse,<br \/>\nTrue,<br \/>\nbnb_4bit_compute_dtype=torch.bfloat16,<br \/>\nbnb_4bit_quant_type=&#8217;nf4&#8242;,<br \/>\nbnb_4bit_use_double_quant=True)<br \/>\nmodel_dir = snapshot_download(&#8220;baichuan-inc\/Baichuan2-13B-Chat&#8221;, revision=&#8217;v1.0.3&#8242;)<br \/>\ntokenizer = AutoTokenizer.from_pretrained(model_dir, device_map=&#8221;auto&#8221;,<br \/>\ntrust_remote_code=True, torch_dtype=torch.float16)<br \/>\nmodel = AutoModelForCausalLM.from_pretrained(model_dir, device_map=&#8221;auto&#8221;,<br \/>\ntrust_remote_code=True, torch_dtype=torch.float16,<br \/>\nquantization_config=quantization_config)<br \/>\nmodel.generation_config = GenerationConfig.from_pretrained(model_dir)<br \/>\nmessages = []<br \/>\nmessages.append({&#8220;role&#8221;: &#8220;user&#8221;, &#8220;content&#8221;: &#8220;\u8bb2\u89e3\u4e00\u4e0b\u201c\u6e29\u6545\u800c\u77e5\u65b0\u201d&#8221;})<br \/>\nresponse = model.chat(tokenizer, messages)<br \/>\nprint(response)<br \/>\nmessages.append({&#8216;role&#8217;: &#8216;assistant&#8217;, &#8216;content&#8217;: response})<br \/>\nmessages.append({&#8220;role&#8221;: &#8220;user&#8221;, &#8220;content&#8221;: &#8220;\u80cc\u8bf5\u4e00\u4e0b\u5c06\u8fdb\u9152&#8221;})<br \/>\nresponse = model.chat(tokenizer, messages)<br \/>\nprint(response)<\/p>\n","protected":false},"excerpt":{"rendered":"<p>import torch from modelscope import snapshot_download,  [&hellip;]<\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[25],"tags":[],"class_list":["post-1362","post","type-post","status-publish","format-standard","hentry","category-ai"],"_links":{"self":[{"href":"https:\/\/www.kumouse.com\/index.php?rest_route=\/wp\/v2\/posts\/1362","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/www.kumouse.com\/index.php?rest_route=\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/www.kumouse.com\/index.php?rest_route=\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/www.kumouse.com\/index.php?rest_route=\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/www.kumouse.com\/index.php?rest_route=%2Fwp%2Fv2%2Fcomments&post=1362"}],"version-history":[{"count":1,"href":"https:\/\/www.kumouse.com\/index.php?rest_route=\/wp\/v2\/posts\/1362\/revisions"}],"predecessor-version":[{"id":1363,"href":"https:\/\/www.kumouse.com\/index.php?rest_route=\/wp\/v2\/posts\/1362\/revisions\/1363"}],"wp:attachment":[{"href":"https:\/\/www.kumouse.com\/index.php?rest_route=%2Fwp%2Fv2%2Fmedia&parent=1362"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/www.kumouse.com\/index.php?rest_route=%2Fwp%2Fv2%2Fcategories&post=1362"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/www.kumouse.com\/index.php?rest_route=%2Fwp%2Fv2%2Ftags&post=1362"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}