{"id":25438,"date":"2025-02-28T17:46:10","date_gmt":"2025-02-28T09:46:10","guid":{"rendered":"http:\/\/139.9.1.231\/?p=25438"},"modified":"2025-06-23T16:57:34","modified_gmt":"2025-06-23T08:57:34","slug":"transformers-generate","status":"publish","type":"post","link":"http:\/\/139.9.1.231\/index.php\/2025\/02\/28\/transformers-generate\/","title":{"rendered":"transformers \u7684 generate() \u65b9\u6cd5\u5b9e\u73b0\u591a\u6837\u5316\u6587\u672c\u751f\u6210\uff1a\u53c2\u6570\u542b\u4e49\u548c\u7b97\u6cd5\u539f\u7406\u89e3\u8bfb"},"content":{"rendered":"\n<p>\u8fd9\u4e2a\u7c7b\u5bf9\u5916\u63d0\u4f9b\u7684\u65b9\u6cd5\u662f&nbsp;<code>generate()<\/code>\uff0c\u901a\u8fc7\u8c03\u53c2\u80fd\u5b8c\u6210\u4ee5\u4e0b\u4e8b\u60c5\uff1a<\/p>\n\n\n\n<ul><li>greedy decoding\uff1a\u5f53\u00a0<code>num_beams=1<\/code>\u00a0\u800c\u4e14\u00a0<code>do_sample=False<\/code>\u00a0\u65f6\uff0c\u8c03\u7528<code>\u00a0greedy_search()<\/code>\u65b9\u6cd5\uff0c\u6bcf\u4e2astep\u751f\u6210\u6761\u4ef6\u6982\u7387\u6700\u9ad8\u7684\u8bcd\uff0c\u56e0\u6b64\u751f\u6210\u5355\u6761\u6587\u672c\u3002<strong><em>\u8d2a\u5a6a\u641c\u7d22\u5bf9\u4e8e\u8f93\u51fa\u76f8\u5bf9\u8f83\u77ed\u4e14\u521b\u9020\u529b\u5e76\u975e\u4f18\u5148\u8003\u8651\u7684\u4efb\u52a1\u975e\u5e38\u6709\u6548\u3002\u7136\u800c\uff0c\u5f53\u751f\u6210\u8f83\u957f\u7684\u5e8f\u5217\u65f6\uff0c\u5b83\u4f1a\u5931\u6548\uff0c\u56e0\u4e3a\u5b83\u4f1a\u5f00\u59cb\u81ea\u6211\u91cd\u590d\u3002<\/em><\/strong><\/li><\/ul>\n\n\n\n<ul><li>multinomial sampling\uff1a\u5f53\u00a0<code>num_beams=1<\/code>\u00a0\u4e14\u00a0<code>do_sample=True<\/code>\u00a0\u65f6\uff0c\u8c03\u7528\u00a0<code>sample()<\/code>\u00a0\u65b9\u6cd5\uff0c\u5bf9\u8bcd\u8868\u505a\u4e00\u4e2a\u91c7\u6837\uff0c\u800c\u4e0d\u662f\u9009\u6761\u4ef6\u6982\u7387\u6700\u9ad8\u7684\u8bcd\uff0c\u589e\u52a0\u591a\u6837\u6027\u3002<\/li><li>beam-search decoding\uff1a<strong>\u5f53\u00a0<code>num_beams>1<\/code>\u00a0\u4e14\u00a0<code>do_sample=False<\/code>\u00a0\u65f6\uff0c\u8c03\u7528\u00a0<code>beam_search()<\/code>\u00a0\u65b9\u6cd5\uff0c\u505a\u4e00\u4e2a num_beams \u7684\u67f1\u641c\u7d22\uff0c\u6bcf\u6b21\u90fd\u662f\u8d2a\u5a6a\u9009\u62e9top N\u4e2a\u67f1\u3002<\/strong><\/li><li>beam-search multinomial sampling\uff1a\u5f53\u00a0<code>num_beams>1<\/code>\u00a0\u4e14\u00a0<code>do_sample=True<\/code>\u00a0\u65f6\uff0c\u8c03\u7528\u00a0<code>beam_sample()<\/code>\u00a0\u65b9\u6cd5\uff0c\u76f8\u5f53\u4e8e\u6bcf\u6b21\u4e0d\u518d\u662f\u8d2a\u5a6a\u9009\u62e9top N\u4e2a\u67f1\uff0c\u800c\u662f\u52a0\u4e86\u4e00\u4e9b\u91c7\u6837\u3002<\/li><li>diverse beam-search decoding\uff1a\u5f53\u00a0<code>num_beams>1<\/code>\u00a0\u4e14\u00a0<code>num_beam_groups>1<\/code>\u00a0\u65f6\uff0c\u8c03\u7528\u00a0<code>group_beam_search()<\/code>\u00a0\u65b9\u6cd5\u3002<\/li><li>constrained beam-search decoding\uff1a\u5f53\u00a0<code>constraints!=None<\/code>\u00a0<strong>\u6216\u8005<\/strong>\u00a0<code>force_words_ids!=None<\/code>\uff0c\u5b9e\u73b0\u53ef\u63a7\u6587\u672c\u751f\u6210\u3002<\/li><\/ul>\n\n\n\n<h2 id=\"h_14481603550_0\">\u53c2\u6570\u5217\u8868<\/h2>\n\n\n\n<blockquote class=\"wp-block-quote\"><p>\u6838\u5fc3\u4ee3\u7801\u8be6\u89c1\uff1a<a href=\"https:\/\/link.zhihu.com\/?target=https%3A\/\/github.com\/huggingface\/transformers\/blob\/main\/src\/transformers\/generation\/utils.py%23L1907C9-L1907C17\" target=\"_blank\" rel=\"noreferrer noopener\">generate()\u5165\u53e3\u51fd\u6570\u5b9a\u4e49<\/a>\uff0c&nbsp;<a href=\"https:\/\/link.zhihu.com\/?target=https%3A\/\/github.com\/huggingface\/transformers\/blob\/main\/src\/transformers\/generation\/configuration_utils.py%23L98\" target=\"_blank\" rel=\"noreferrer noopener\">GenerationConfig\u7c7b<\/a><\/p><\/blockquote>\n\n\n\n<h3 id=\"h_14481603550_1\"><strong>1.\u63a7\u5236\u751f\u6210\u957f\u5ea6\u7684\u53c2\u6570<\/strong><\/h3>\n\n\n\n<figure class=\"wp-block-table\"><table><tbody><tr><td>\u53c2\u6570<\/td><td>\u7c7b\u578b<\/td><td>\u7f3a\u7701\u503c<\/td><td>\u542b\u4e49<\/td><\/tr><tr><td>max_length<\/td><td>int<\/td><td>20<\/td><td>\u8868\u793a prompt + max_new_tokens \u7d2f\u52a0\u7684\u6700\u5927\u957f\u5ea6\uff0c\u5982\u679cmax_new_tokens\u4e5f\u8bbe\u7f6e\u4e86\uff0c\u4f1a\u8986\u76d6\u8fd9\u4e2a\u53c2\u6570<\/td><\/tr><tr><td>max_new_tokens<\/td><td>int<\/td><td><\/td><td>\u751f\u6210\u90e8\u5206\u7684tokens\u7684\u6700\u5927\u957f\u5ea6 \uff08\u5ffd\u7565prompt\u90e8\u5206\u7684\u957f\u5ea6\uff09<\/td><\/tr><tr><td>min_length<\/td><td><\/td><td>0<\/td><td>\u8868\u793a prompt + min_new_tokens \u7d2f\u52a0\u7684\u6700\u5c0f\u957f\u5ea6\uff0c\u5982\u679cmin_new_tokens\u4e5f\u8bbe\u7f6e\u4e86\uff0c\u4f1a\u8986\u76d6\u8fd9\u4e2a\u53c2\u6570<\/td><\/tr><tr><td>min_new_tokens<\/td><td>int<\/td><td><\/td><td>\u751f\u6210\u90e8\u5206\u7684tokens\u7684\u6700\u5c0f\u957f\u5ea6 \uff08\u5ffd\u7565prompt\u90e8\u5206\u7684\u957f\u5ea6\uff09<\/td><\/tr><tr><td>early_stopping<\/td><td>bool, str<\/td><td>False<\/td><td>\u5bf9\u4e8e<a href=\"https:\/\/zhida.zhihu.com\/search?content_id=251919076&amp;content_type=Article&amp;match_order=1&amp;q=beam+search&amp;zhida_source=entity\" target=\"_blank\" rel=\"noreferrer noopener\">beam search<\/a>\u65b9\u6cd5\u7684\u63a7\u5236\u7ec8\u6b62\u7684\u914d\u7f6e\u3002<br>False: \u5f53\u6709&#8217;num_beams&#8217;\u4e2a\u5019\u9009\u751f\u6210\uff0c\u5219\u7ec8\u6b62<br>True: \u5e94\u7528\u4e00\u4e9b\u542f\u53d1\u5f0f\u89c4\u5219\u5224\u65ad\u4e0d\u80fd\u627e\u5230\u66f4\u597d\u7684\u751f\u6210\u5019\u9009\uff0c\u6765\u63d0\u524d\u7ec8\u6b62\u751f\u6210<br>&#8220;never&#8221;\uff1a \u5f53\u5224\u65ad\u6ca1\u6709\u66f4\u597d\u7684\u53ef\u751f\u6210\u7684candidate, beam search \u8fc7\u7a0b\u7ec8\u6b62<\/td><\/tr><tr><td>max_time<\/td><td>float<\/td><td><\/td><td>\u6267\u884c\u751f\u6210\u7684\u6700\u5927\u65f6\u95f4(s\u79d2\u6570)<\/td><\/tr><tr><td>stop_strings<\/td><td>str, array[str]<\/td><td><\/td><td>\u914d\u7f6e\u6a21\u578b\u751f\u6210\u7684\u7ec8\u6b62\u5b57\u7b26\u4e32\uff0c\u5f53\u6a21\u578b\u751f\u6210\u53c2\u6570\u914d\u7f6e\u7684\u5b57\u7b26\u4e32\uff0c\u5219\u7ec8\u6b62\u751f\u6210\u3002<\/td><\/tr><\/tbody><\/table><\/figure>\n\n\n\n<h3 id=\"h_14481603550_2\">2. \u63a7\u5236\u751f\u6210\u7b56\u7565\u7684\u53c2\u6570<\/h3>\n\n\n\n<figure class=\"wp-block-table\"><table><tbody><tr><td>\u53c2\u6570<\/td><td>\u7c7b\u578b<\/td><td>\u7f3a\u7701\u503c<\/td><td>\u542b\u4e49<\/td><\/tr><tr><td>do_sample<\/td><td>bool<\/td><td>False<\/td><td><strong>True: \u751f\u6210\u8fc7\u7a0b\u4f7f\u7528\u91c7\u6837\u903b\u8f91<br>False: \u4f7f\u7528greedy\u505a\u751f\u6210<\/strong><br><em>\u82e5 <code>do_sample=False<\/code>\uff08\u9ed8\u8ba4\uff09\uff0c\u7cfb\u7edf\u6309\u8d2a\u5fc3\u6216 beam search \u7b56\u7565\u9009\u62e9 <strong>\u6700\u9ad8\u6982\u7387\u7684\u90a3\u4e2a token<\/strong>\uff1b\u82e5 <code>do_sample=True<\/code>\uff0c\u5728\u6bcf\u4e00\u6b65\u751f\u6210\u65f6\uff0c\u4f1a\u6839\u636e\u6a21\u578b\u8f93\u51fa\u7684\u6982\u7387\u5206\u5e03<strong>\u968f\u673a\u91c7\u6837<\/strong>\u4e0b\u4e00\u4e2a token\uff0c\u800c\u4e0d\u662f\u603b\u662f\u9009\u62e9\u6700\u9ad8\u6982\u7387\u9879<\/em><\/td><\/tr><tr><td>num_beams<\/td><td>int<\/td><td>1<\/td><td>\u8bbe\u7f6ebeam search \u675f\u7684\u6570\u91cf\u3002\u5982\u679c\u662f1\u4e0d\u505abeam search \u641c\u7d22\u3002<br><em><code>num_beams = N<\/code> \u8868\u793a\u5728\u751f\u6210\u8fc7\u7a0b\u4e2d <strong>\u540c\u65f6\u4fdd\u7559 N \u6761\u5019\u9009\u5e8f\u5217<\/strong>\uff08\u201cbeams\u201d\uff09<\/em><\/td><\/tr><tr><td>num_beam_groups<\/td><td>int<\/td><td>1<\/td><td>\u4e3a\u4e86\u4fdd\u8bc1\u751f\u6210\u7684\u591a\u6837\u6027\uff0c\u5c06num_beams \u8bbe\u7f6e\u6210\u591a\u7ec4\u3002\u53c2\u8003\u6587\u732e\uff1a&nbsp;<a href=\"https:\/\/link.zhihu.com\/?target=https%3A\/\/arxiv.org\/pdf\/1610.02424.pdf\" target=\"_blank\" rel=\"noreferrer noopener\">https:\/\/arxiv.org\/pdf\/1610.02424.pdf<\/a><\/td><\/tr><tr><td>penalty_alpha<\/td><td>float<\/td><td><\/td><td>contrastive search decoding\u7684\u914d\u7f6e\u9879\uff0c\u7528\u4e8e\u5e73\u8861\u751f\u6210\u7f6e\u4fe1\u5ea6\u548c\u8870\u51cf\u7684\u60e9\u7f5a<\/td><\/tr><tr><td>dola_layers<\/td><td>str, List[int]<\/td><td><\/td><td>str :<br>&#8220;None&#8221;\uff1a \u4e0d\u4f7f\u7528dola<br>&#8220;low&#8221; : \u8f83\u4f4e\u7684\u4e00\u534alayers, \u6700\u591a20\u5c42\u4f7f\u7528dola<br>&#8220;high&#8221;: \u8f83\u9ad8\u7684\u4e00\u534alayers, \u6700\u591a20\u5c42\u4f7f\u7528dola<br>List[int] : \u901a\u8fc7\u6307\u5b9a\u4e00\u4e2aindex\u6570\u7ec4\uff0c\u6307\u5b9adola \u5c42<br>&#8220;low&#8221;\uff1a \u63d0\u5347\u957f\u7b54\u6848\u7684task\uff0c<br>&#8220;high&#8221;\uff1a\u63d0\u5347\u77ed\u7b54\u6848\u7684task<\/td><\/tr><\/tbody><\/table><\/figure>\n\n\n\n<h3 id=\"h_14481603550_3\">3.cache\u914d\u7f6e\u53c2\u6570<\/h3>\n\n\n\n<figure class=\"wp-block-table\"><table><tbody><tr><td>\u53c2\u6570<\/td><td>\u7c7b\u578b<\/td><td>\u7f3a\u7701\u503c<\/td><td>\u542b\u4e49<\/td><\/tr><tr><td>use_cache<\/td><td>bool<\/td><td>True<\/td><td>\u662f\u5426\u4f7f\u7528<a href=\"https:\/\/zhida.zhihu.com\/search?content_id=251919076&amp;content_type=Article&amp;match_order=1&amp;q=KV+cache&amp;zhida_source=entity\" target=\"_blank\" rel=\"noreferrer noopener\">KV cache<\/a>&nbsp;\u52a0\u901f\u63a8\u7406\u901f\u5ea6<\/td><\/tr><tr><td>cache_implementation<\/td><td>str<\/td><td><\/td><td>\u6307\u5b9acache\u5b9e\u73b0\u7684name\uff0c\u5728\u8c03\u7528generate()\u65f6\uff0c\u5b9e\u4f8b\u5316cache\u3002<br>\ufeff&#8221;static&#8221;: [StaticCache]<br>&#8220;offloaded_static&#8221;: [OffloadedStaticCache]<br>\ufeff&#8221;sliding_window&#8221;: [SlidingWindowCache]\ufeff<br>&#8220;hybrid&#8221;: [HybridCache]\ufeff<br>&#8220;mamba&#8221;: [MambaCache]<br>\ufeff&#8221;quantized&#8221;:[QuantizedCache]<\/td><\/tr><tr><td>cache_config<\/td><td>CacheConfig \uff0c dict<\/td><td>None<\/td><td>cache\u7c7b\u4f7f\u7528\u7684\u53c2\u6570<\/td><\/tr><tr><td>return_legacy_cache<\/td><td>bool<\/td><td>True<\/td><td>\u5f53DynamicCache \u88ab\u4f7f\u7528\u65f6\uff0c\u662f\u5426\u8fd4\u56de\u5386\u53f2\u7684\u548c\u65b0\u683c\u5f0f\u7684cache<\/td><\/tr><\/tbody><\/table><\/figure>\n\n\n\n<h3 id=\"h_14481603550_4\">4.\u64cd\u4f5c\u6a21\u578b\u8f93\u51falogit\u7684\u914d\u7f6e\u53c2\u6570<\/h3>\n\n\n\n<figure class=\"wp-block-table\"><table><tbody><tr><td>\u53c2\u6570<\/td><td>\u7c7b\u578b<\/td><td>\u7f3a\u7701\u503c<\/td><td>\u542b\u4e49<\/td><\/tr><tr><td>temperature<\/td><td>float<\/td><td>1.0<\/td><td>\u8fd9\u4e2a\u503c\u7528\u4e8e\u5efa\u6a21\u4e0b\u4e00\u4e2atoken\u7684\u6982\u7387\uff0c \u8fd9\u4e2a\u503c\u88ab\u8bbe\u7f6e\u5728generation_config.json\u6587\u4ef6\u4e2d<br>\u4ec5\u5728 <strong><code>do_sample=True<\/code> \u65f6<\/strong>\u751f\u6548\uff0c<strong>\u63a7\u5236 logits \u7ecf\u8fc7 softmax \u540e\u5206\u5e03\u7684\u5e73\u6ed1\/\u5c16\u9510\u7a0b\u5ea6<\/strong><\/td><\/tr><tr><td>top_k<\/td><td>int<\/td><td>50<\/td><td>\u7b5b\u9009\u6700\u9ad8\u6982\u7387\u7684top k\u4e2a\u8bcd\uff0c \u8fd9\u4e2a\u503c\u88ab\u8bbe\u7f6e\u5728generation_config.json\u6587\u4ef6\u4e2d<\/td><\/tr><tr><td>top_p<\/td><td>float<\/td><td>1.0<\/td><td>\u5f53\u8bbe\u7f6e&lt;1\u65f6\uff0c\u7b5b\u9009\u6982\u7387\u6700\u9ad8\u7684token\uff0c\u7d2f\u52a0\u6982\u7387\u4e0d\u8d85\u8fc7top_p\u7684token<\/td><\/tr><tr><td>min_p<\/td><td>float<\/td><td><\/td><td>\u914d\u7f6e\u7b5b\u9009\u6982\u7387\u6700\u4f4e\u7684\u4e00\u6279token, \u7d2f\u52a0\u6982\u7387\u4e0d\u8d85\u8fc7min_p\uff0c\u88c1\u526a\u6389\uff0c\u8be5\u914d\u7f6e\u76f8\u5f53\u4e8etop_p\u7684\u53cd\u5411\u64cd\u4f5c<\/td><\/tr><tr><td>typical_p<\/td><td>float<\/td><td>1.0<\/td><td>\u6d4b\u91cf\u4e24\u4e2a\u5206\u5e03\u7684\u76f8\u4f3c\u6027\uff1a \u9884\u6d4b\u4e0b\u4e00\u4e2a\u76ee\u6807token\u7684\u6982\u7387 and \u9884\u6d4b\u4e0b\u4e00\u4e2a\u968f\u673aToken\u7684\u6761\u4ef6\u6982\u7387\u671f\u671b\u3002\u5982\u679c\u8bbe\u7f6e&lt;1\uff0c\u5219\u7b5b\u9009\u6700\u5178\u578b\u7684token\u3002<\/td><\/tr><tr><td>epsilon_cutoff<\/td><td>float<\/td><td>0.0<\/td><td>\u6309\u8bbe\u7f6e\u7684\u503c\uff0c\u5361\u6389\u4f4e\u6982\u7387\u503c\u7684token\uff0c\u4e00\u822c\u8bbe\u7f6e\u4e3a\uff1a3e-4 to 9e-4<\/td><\/tr><tr><td>eta_cutoff<\/td><td>float<\/td><td>0.0<\/td><td>\u6df7\u5408\u5c40\u90e8\u5178\u578b\u6027\u91c7\u6837\u548cepsilon\u91c7\u6837\u65b9\u6cd5<\/td><\/tr><tr><td>diversity_penalty<\/td><td>float<\/td><td>0.0<\/td><td>\u53ea\u5bf9group beam search\u65b9\u6cd5\u751f\u6548\uff0c\u5982\u679c\u5728\u67d0\u4e2a\u7279\u5b9a\u65f6\u95f4\u751f\u6210\u7684token\u4e0e\u4efb\u4f55beam \u7ec4\u751f\u6210\u7684token\u4e00\u81f4\uff0c\u5219beam\u7684score\u51cf\u53bb\u8fd9\u4e2a\u503c<\/td><\/tr><tr><td>repetition_penalty<\/td><td>float<\/td><td>1.0<\/td><td>1.0 \u9ed8\u8ba4\u4e0d\u60e9\u7f5a<\/td><\/tr><tr><td>encoder_repetition_penalty<\/td><td>float<\/td><td>1.0<\/td><td>\u5bf9\u4e8e\u4e0d\u5728\u539f\u59cb\u8f93\u5165\u7684token\uff0c\u6307\u6570\u7ea7\u7684\u60e9\u7f5a<\/td><\/tr><tr><td>length_penalty<\/td><td>float<\/td><td>1.0<\/td><td>\u5bf9\u4e8ebeam \u7c7b\u7684\u751f\u6210\u65b9\u6cd5\u7684\u957f\u5ea6\u60e9\u7f5a\uff0c\u7531\u4e8e\u5e8f\u5217score\u662f log likelihood \uff0c &gt; 0 \u503e\u5411\u4e8e\u66f4\u957f\u7684 &lt;0 \u503e\u5411\u4e8e\u66f4\u77ed\u7684<\/td><\/tr><tr><td>no_repeat_ngram_size<\/td><td>int<\/td><td>0<\/td><td>\u5982\u679c\u5927\u4e8e0\uff0c \u5219\u5bf9\u5e94\u7684size\u7684ngram\u53ea\u80fd\u51fa\u73b01\u6b21<\/td><\/tr><tr><td>bad_words_ids<\/td><td>List[List[int]]<\/td><td><\/td><td>\u5217\u51fa\u4e0d\u5141\u8bb8\u751f\u6210\u7684tokens_id<\/td><\/tr><tr><td>force_words_ids<\/td><td>List[List[int]] or List[List[List[int]]]<\/td><td><\/td><td>\u5fc5\u987b\u88ab\u751f\u6210\u7684words_ids\u3002 \u5982\u679c\u914d\u7f6eList[List[List[int]]] \u8bbe\u7f6e\u5bf9\u4e8e\u6bcf\u4e2atoken\u7684\u7ea6\u675f<\/td><\/tr><tr><td>renormalize_logits<\/td><td>bool<\/td><td>False<\/td><td>\u5bf9\u4e8e\u6240\u6709\u7684logits\u505a\u540e\u5904\u7406\u540e\uff0c\u662f\u5426\u8981\u518d\u505a\u4e0bnormalize<\/td><\/tr><tr><td>constraints<\/td><td>List[Constraint]<\/td><td><\/td><td>\u901a\u8fc7\u5b9a\u4e49\u4e00\u4e2aList[Constraint] \u5bf9\u8c61\u6570\u7ec4\uff0c\u6765\u786e\u4fdd\u8f93\u51fa\u662f\u5728\u67d0\u4e9b\u9650\u5236\u7684\u573a\u666f\u4e0b\u3002\u4e00\u822c\u7528\u4e8e\u5b89\u5168\u7684\u573a\u666f<\/td><\/tr><tr><td>forced_bos_token_id<\/td><td>int<\/td><td>model.config.forced_bos_token_id<\/td><td>\u5f3a\u5236\u8ddf\u5728decoder_start_token_id\u4e4b\u540e\u7684\u7b2c\u4e00\u4e2atoken\uff0c\u5bf9\u591a\u8bed\u8a00\u6a21\u578b\u662f\u6709\u7528\u7684<\/td><\/tr><tr><td>forced_eos_token_id<\/td><td>int or List[int]<\/td><td>model.config.forced_eos_token_id<\/td><td>\u5f53\u751f\u6210\u7684token\u8fbe\u5230max_length\u4e0a\u9650\u65f6\uff0c\u6700\u540e\u4e00\u4f4d\u8f93\u51fa\u7684token<\/td><\/tr><tr><td>remove_invalid_values<\/td><td>bool<\/td><td>model.config.remove_invalid_values<\/td><td>\u662f\u5426\u79fb\u51fa\u53ef\u80fd\u751f\u6210\u7684nan and inf \u503c\uff0c\u914d\u7f6e\u8fd9\u4e2a\u4f1a\u51cf\u6162\u751f\u6210\u901f\u5ea6<\/td><\/tr><tr><td>exponential_decay_length_penalty<\/td><td>tuple(int, float)<\/td><td><\/td><td>\u6307\u6570\u7ea7\u589e\u52a0\u957f\u5ea6\u7684\u60e9\u7f5a\uff0ctuple\uff08start_index, decay_factor\uff09 start index \u6307\u793a\u60e9\u7f5a\u7684\u5f00\u59cbi\uff0cdecay_factor \u6307\u6570\u8870\u51cf\u7684\u60e9\u7f5a\u56e0\u5b50<\/td><\/tr><tr><td>suppress_tokens<\/td><td>List[int]<\/td><td><\/td><td>\u901a\u8fc7\u8bbe\u7f6e\u7981\u6b62\u7684token\u7684logit\u4e3a-inf\uff0c\u6765\u7981\u6b62token\u88absample<\/td><\/tr><tr><td>begin_suppress_tokens<\/td><td>List[int]<\/td><td><\/td><td>\u901a\u8fc7\u8bbe\u7f6e\u9996\u4f4d\u7981\u6b62\u7684token\u7684logit\u4e3a-inf\uff0c\u6765\u7981\u6b62\u9996\u4f4d\u8fd9\u90e8\u5206token\u88ab\u91c7\u6837\u5230\uff0c\u8fdb\u800c\u5bfc\u81f4\u88ab\u751f\u6210<\/td><\/tr><tr><td>forced_decoder_ids<\/td><td>List[List[int]]<\/td><td><\/td><td>\u4e00\u4e2a\u6574\u6570pair\u7684\u6570\u7ec4\uff0c\u683c\u5f0f[\u751f\u6210index\uff0c token_index]\u6307\u793a\u56fa\u5b9a\u4f4d\u7f6e\u5f3a\u5236\u751f\u6210\u67d0\u4e2atoken\uff0c\u4f8b\u5982[[1, 123]] \u7b2c\u4e8c\u4e2a\u4f4d\u7f6e\u603b\u662f\u751f\u6210token 123<\/td><\/tr><tr><td>sequence_bias<\/td><td>Dict[Tuple[int], float]<\/td><td><\/td><td>token list -&gt; bias\u7684\u6620\u5c04\uff0c\u6b63\u7684bias\u63d0\u5347\u51e0\u7387\uff0c\u8d1f\u7684bias\u964d\u4f4e\u51e0\u7387<\/td><\/tr><tr><td>token_healing<\/td><td>bool<\/td><td>False<\/td><td>\u5bf9prompt\u5c3e\u90e8\u7684token\u505a\u76f8\u4f3c\u66ff\u6362\uff0c\u4ee5\u63d0\u5347\u751f\u6210\u8d28\u91cf<\/td><\/tr><tr><td>guidance_scale<\/td><td>float<\/td><td><\/td><td>\u662f\u4e00\u4e2a\u7f29\u653e\u56e0\u5b50\uff0c\u5f53&gt;1\u65f6\uff0c\u8fd9\u4e2a\u56e0\u5b50\u8d8a\u9ad8\uff0c\u8d8a\u9f13\u52b1\u6a21\u578b\u751f\u6210\u4e0eprompt\u63a5\u8fd1\u7684samples \u3002<\/td><\/tr><tr><td>watermarking_config<\/td><td>BaseWatermarkingConfig or dict<\/td><td><\/td><td>\u5bf9\u8f93\u51fa\u7ed3\u679c\u589e\u52a0\u6c34\u5370<\/td><\/tr><\/tbody><\/table><\/figure>\n\n\n\n<h3 id=\"h_14481603550_5\">5.\u8f93\u51fa\u7ed3\u679c\u914d\u7f6e\u53c2\u6570<\/h3>\n\n\n\n<figure class=\"wp-block-table\"><table><tbody><tr><td>\u53c2\u6570<\/td><td>\u7c7b\u578b<\/td><td>\u7f3a\u7701\u503c<\/td><td>\u542b\u4e49<\/td><\/tr><tr><td>num_return_sequences<\/td><td>int<\/td><td>1<\/td><td>\u5bf9\u4e8ebatch\u4e2d\u7684\u6bcf\u4e2a\u5143\u7d20\uff0c\u8bbe\u7f6e\u72ec\u7acb\u8ba1\u7b97\u7684\u8fd4\u56de\u7684sequence\u7684\u6570\u91cf<\/td><\/tr><tr><td>output_attentions<\/td><td>bool<\/td><td>False<\/td><td>\u662f\u5426\u8fd4\u56de\u6240\u6709\u7684attention\u7684\u5411\u91cf<\/td><\/tr><tr><td>output_hidden_states<\/td><td>bool<\/td><td>False<\/td><td>\u662f\u5426\u8fd4\u56de\u6240\u6709\u7f51\u7edc\u5c42\u7684\u9690\u5c42\u72b6\u6001<\/td><\/tr><tr><td>output_scores<\/td><td>bool<\/td><td>False<\/td><td>\u662f\u5426\u8fd4\u56deprediction scores<\/td><\/tr><tr><td>output_logits<\/td><td>bool<\/td><td><\/td><td>\u662f\u5426\u8fd4\u56de\u672a\u5904\u7406\u8fc7\u7684\u7684logit score<\/td><\/tr><tr><td>return_dict_in_generate<\/td><td>bool<\/td><td>False<\/td><td>\u9664\u4e86\u8fd4\u56de\u751f\u6210\u5e8f\u5217\uff0c\u662f\u5426\u8fd8\u8fd4\u56dea [`~utils.ModelOutput`]<\/td><\/tr><\/tbody><\/table><\/figure>\n\n\n\n<h3 id=\"h_14481603550_6\">6.\u751f\u6210\u65f6\u4f7f\u7528\u7684\u7279\u6b8atoken\u7684\u914d\u7f6e\u53c2\u6570<\/h3>\n\n\n\n<figure class=\"wp-block-table\"><table><tbody><tr><td>\u53c2\u6570<\/td><td>\u7c7b\u578b<\/td><td>\u7f3a\u7701\u503c<\/td><td>\u542b\u4e49<\/td><\/tr><tr><td>pad_token_id<\/td><td>int<\/td><td><\/td><td>padding token ID<\/td><\/tr><tr><td>bos_token_id<\/td><td>int<\/td><td><\/td><td>beginning -of &#8211; sequence token ID<\/td><\/tr><tr><td>eos_token_id<\/td><td>Union[int, List[int]]<\/td><td><\/td><td>end-of-sequence token ID<\/td><\/tr><\/tbody><\/table><\/figure>\n\n\n\n<h3 id=\"h_14481603550_7\">6.\u8f85\u52a9\u751f\u6210\u7684\u914d\u7f6e\u53c2\u6570\uff08\u6295\u673a\u91c7\u6837\uff09<\/h3>\n\n\n\n<figure class=\"wp-block-table\"><table><tbody><tr><td>\u53c2\u6570<\/td><td>\u7c7b\u578b<\/td><td>\u7f3a\u7701\u503c<\/td><td>\u542b\u4e49<\/td><\/tr><tr><td>is_assistant<\/td><td>bool<\/td><td>False<\/td><td>\u6307\u5b9a\u662f\u5426\u6a21\u578b\u662f\u4e00\u4e2aassistant(draft) model<\/td><\/tr><tr><td>num_assistant_tokens<\/td><td>int<\/td><td>20<\/td><td>\u6295\u673a\u91c7\u6837\u8fc7\u7a0b\uff0c\u6bcf\u6b21\u8fed\u4ee3&nbsp;<a href=\"https:\/\/zhida.zhihu.com\/search?content_id=251919076&amp;content_type=Article&amp;match_order=1&amp;q=assistant+model&amp;zhida_source=entity\" target=\"_blank\" rel=\"noreferrer noopener\">assistant model<\/a>&nbsp;\u8981\u8f93\u51fa\u591a\u5c11\u4e2atoken\uff0c\u7ed9\u5230\u76ee\u6807\u6a21\u578b\u505acheck\u3002\u914d\u7f6e\u66f4\u9ad8\u7684\u503c\uff0c\u5982\u679cassistant model \u6548\u679c\u597d \u80fd\u5e26\u6765\u66f4\u597d\u7684\u52a0\u901f\u6bd4<\/td><\/tr><tr><td>num_assistant_tokens_schedule<\/td><td>str<\/td><td>constant<\/td><td>&#8220;heuristic&#8221; \uff1a \u5f53\u6240\u6709\u6295\u673a\u91c7\u6837\u7684token\u90fd\u6b63\u786e\u65f6\uff0c\u5c06num_assistant_tokens\u589e\u52a02\uff0c\u5426\u5219\u51cf\u5c111\u3002\ufeff<br>&#8220;constant&#8221;: num_assistant_tokens \u4fdd\u6301\u56fa\u5b9a\u4e0d\u53d8\ufeff<br>&#8220;heuristic_transient&#8221;\uff1a \u7c7b\u4f3c\u4e8e\u542f\u53d1\u5f0f\u65b9\u6cd5\uff0c\u6bcf\u6b21\u751f\u6210\u8c03\u7528\uff0c\u90fd\u7f6e\u6210\u521d\u59cb\u5316\u7684num_assistant_tokens\u503c<\/td><\/tr><tr><td>assistant_confidence_threshold<\/td><td>float<\/td><td>0.4<\/td><td>\u5f53assistant model\u9884\u4f30\u5f53\u524dtoken\u7684\u7f6e\u4fe1\u5ea6 \u5c0f\u4e8e \u9608\u503c\u65f6\uff0c\u63d0\u524d\u7ec8\u6b62assistant model\u7684\u751f\u6210<\/td><\/tr><tr><td>prompt_lookup_num_tokens<\/td><td>int<\/td><td><\/td><td>\u4f5c\u4e3a\u5019\u9009token \u8981\u8f93\u51fa\u7684token\u7684\u6570\u91cf<\/td><\/tr><tr><td>max_matching_ngram_size<\/td><td>int<\/td><td>2<\/td><td>match prompt\u7684\u6700\u5927ngram\u7684\u6570\u91cf<\/td><\/tr><tr><td>assistant_early_exit<\/td><td>int<\/td><td><\/td><td><\/td><\/tr><tr><td>assistant_lookbehind<\/td><td>int<\/td><td>10<\/td><td>\u5982\u679c\u8bbe\u7f6e\u4e3a\u6b63\u6574\u6570\uff0c\u5219\u91cd\u65b0\u7f16\u7801\u8fc7\u7a0b\u5c06\u989d\u5916\u8003\u8651\u6700\u540e\u7684assistant_lookbehind\u4e2a\u8f85\u52a9\u6807\u8bb0\uff0c\u4ee5\u6b63\u786e\u5bf9\u9f50\u6807\u8bb0\u3002\u6b64\u8bbe\u7f6e\u4ec5\u53ef\u5728\u63a8\u6d4b\u89e3\u7801\u4e2d\u4f7f\u7528\u4e0d\u540c\u7684\u5206\u8bcd\u5668\u65f6\u4f7f\u7528\u3002<\/td><\/tr><tr><td>target_lookbehind<\/td><td>int<\/td><td>10<\/td><td>\u5982\u679c\u8bbe\u7f6e\u4e3a\u6b63\u6574\u6570\uff0c\u5219\u91cd\u65b0\u7f16\u7801\u8fc7\u7a0b\u5c06\u989d\u5916\u8003\u8651\u6700\u540e\u7684target_lookbehind\u4e2a\u8f85\u52a9\u6807\u8bb0\uff0c\u4ee5\u6b63\u786e\u5bf9\u9f50\u6807\u8bb0\u3002\u6b64\u8bbe\u7f6e\u4ec5\u53ef\u5728\u63a8\u6d4b\u89e3\u7801\u4e2d\u4f7f\u7528\u4e0d\u540c\u7684\u5206\u8bcd\u5668\u65f6\u4f7f\u7528\u3002<\/td><\/tr><\/tbody><\/table><\/figure>\n\n\n\n<p><br>\u5982\u6709\u6574\u7406\u9519\u8bef\uff0c\u6b22\u8fce\u6307\u6b63\uff5e<\/p>\n","protected":false},"excerpt":{"rendered":"<p>\u8fd9\u4e2a\u7c7b\u5bf9\u5916\u63d0\u4f9b\u7684\u65b9\u6cd5\u662f&nbsp;generate()\uff0c\u901a\u8fc7\u8c03\u53c2\u80fd\u5b8c\u6210\u4ee5\u4e0b\u4e8b\u60c5\uff1a greedy decodin &hellip; <a href=\"http:\/\/139.9.1.231\/index.php\/2025\/02\/28\/transformers-generate\/\" class=\"more-link\">\u7ee7\u7eed\u9605\u8bfb<span class=\"screen-reader-text\">transformers \u7684 generate() \u65b9\u6cd5\u5b9e\u73b0\u591a\u6837\u5316\u6587\u672c\u751f\u6210\uff1a\u53c2\u6570\u542b\u4e49\u548c\u7b97\u6cd5\u539f\u7406\u89e3\u8bfb<\/span><\/a><\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":[],"categories":[11,4,38],"tags":[],"_links":{"self":[{"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/posts\/25438"}],"collection":[{"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/comments?post=25438"}],"version-history":[{"count":13,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/posts\/25438\/revisions"}],"predecessor-version":[{"id":27167,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/posts\/25438\/revisions\/27167"}],"wp:attachment":[{"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/media?parent=25438"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/categories?post=25438"},{"taxonomy":"post_tag","embeddable":true,"href":"http:\/\/139.9.1.231\/index.php\/wp-json\/wp\/v2\/tags?post=25438"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}