{"id":2666,"date":"2023-05-13T21:13:18","date_gmt":"2023-05-13T13:13:18","guid":{"rendered":"https:\/\/divis.cn\/?p=2666"},"modified":"2026-02-07T20:13:20","modified_gmt":"2026-02-07T12:13:20","slug":"2666","status":"publish","type":"post","link":"https:\/\/www.divis.cn\/index.php\/2023\/05\/13\/2666\/","title":{"rendered":"Taiyi-Stable-Diffusion Finetune\u793a\u4f8b\u7684\u4f7f\u7528"},"content":{"rendered":"

\u8fd0\u884c\u5bf9\u8c61\uff1ahttps:\/\/github.com\/IDEA-CCNL\/Fengshenbang-LM\/tree\/main\/fengshen\/examples\/finetune_taiyi_stable_diffusion<\/a><\/p>\n

\u5b89\u88c5\u73af\u5883\uff1a\u4e00\u53f0\u62e5\u6709\u516b\u53f03090\u663e\u5361\u7684\u670d\u52a1\u5668\u4e2d\u5185\u5efa\u7684linux ubuntu 20.04\u865a\u62df\u673a\u3002\u56e0\u4e3a\u8fd9\u4e2a\u670d\u52a1\u5668\u8fd8\u6709\u5f88\u591a\u5176\u4ed6\u4eba\u8981\u7528\u6240\u4ee5\u8d26\u53f7\u6743\u9650\u8f83\u4f4e\u3002\u4f7f\u7528\u767d\u540d\u5355\u673a\u5236\u7684ssh\u534f\u8bae\u8fde\u63a5\u3002<\/h4>\n

\u7b2c\u4e00\u6b65\uff1a\u914d\u7f6e\u672c\u673assh\u73af\u5883\u4ee5\u4fbf\u8fde\u63a5\u670d\u52a1\u5668<\/h2>\n

\u5148\u5728\u672c\u673awindows\u73af\u5883\u4e2d\u5b89\u88c5OpenSSH\u534f\u8bae\uff0c\u7136\u540e\u4f7f\u7528vscode\u914d\u7f6essh\u3002\u914d\u7f6e\u597d\u4ee5\u540e\u4f7f\u7528ssh\u547d\u4ee4\u901a\u8fc7powershell\u76f4\u63a5\u8bbf\u95ee\u8fdc\u7a0b\u670d\u52a1\u5668\u3002\u5177\u4f53\u6d41\u7a0b\u53c2\u8003\uff1a<\/strong>https:\/\/blog.csdn.net\/qq_45956730\/article\/details\/124277747<\/a>
\n\u914d\u7f6e\u597d\u4ee5\u540e\u76f4\u63a5\u5728powershell\u91cc\u9762\u8f93\u5165ssh g231\u5c31\u53ef\u4ee5\u8bbf\u95ee\u8fdc\u7a0b\u4e3b\u673a\u3002
\n\u8fd9\u91cc\u4e0d\u4f7f\u7528xshell\u3001finalshell\u4e4b\u7c7b\u7684\u8fdc\u7a0b\u8fde\u63a5\u8f6f\u4ef6\u662f\u56e0\u4e3a\u6d59\u5927\u8fd9\u8fb9\u7684\u535a\u58eb\u90fd\u662f\u76f4\u63a5\u901a\u8fc7\u547d\u4ee4\u884c\u8bbf\u95ee\u4e3b\u673a\u7684\uff0c\u6211\u4e5f\u53ea\u597d\u5165\u4e61\u968f\u4fd7\u4e86\u3002\u800c\u4e14vscode\u65e2\u7136\u53c8\u80fd\u7f16\u7a0b\u53c8\u80fd\u5904\u7406ssh\u914d\u7f6e\u90a3\u4e0d\u59a8\u90fd\u7528\u5b83\u6765\u5904\u7406\u4ee3\u7801\u76f8\u5173\u7684\u5de5\u4f5c\u3002\u800c\u4e14\uff0cvscode\u8fd8\u53ef\u4ee5\u76f4\u63a5\u7f16\u8f91\u8fdc\u7a0b\u4e3b\u673a\u4e0a\u7684\u6587\u4ef6\uff0c\u6bd4vim\u65b9\u4fbf\u591a\u4e86\uff1a<\/p>\n

\u7b2c\u4e8c\u6b65\uff1a\u5b89\u88c5\u592a\u4e59<\/h2>\n

2.1 \u57fa\u672c\u6d41\u7a0b<\/h3>\n

\u4e3b\u8981\u53c2\u8003\u6587\u6863\uff1ahttps:\/\/github.com\/IDEA-CCNL\/Fengshenbang-LM#%E5%AE%89%E8%A3%85<\/a>
\n\u91c7\u7528\u672c\u5730\u5b89\u88c5\u65b9\u5f0f\uff1a<\/p>\n

git clone https:\/\/github.com\/IDEA-CCNL\/Fengshenbang-LM.git<\/a>
\ncd Fengshenbang-LM
\ngit submodule init
\ngit submodule update<\/p>\n

submodule\u662f\u6211\u4eec\u7528\u6765\u7ba1\u7406\u6570\u636e\u96c6\u7684fs_datasets\uff0c\u901a\u8fc7ssh\u7684\u65b9\u5f0f\u62c9\u53d6\uff0c\u5982\u679c\u7528\u6237\u6ca1\u6709\u5728\u673a\u5668\u4e0a\u914d\u7f6essh-key\u7684\u8bdd\u53ef\u80fd\u4f1a\u62c9\u53d6\u5931\u8d25<\/p>\n

\u5982\u679c\u62c9\u53d6\u5931\u8d25\uff0c\u9700\u8981\u5230.gitmodules\u6587\u4ef6\u4e2d\u628assh\u5730\u5740\u6539\u4e3ahttps\u5730\u5740\u5373\u53ef\u3002<\/p>\n

pip install –editable .<\/p><\/blockquote>\n

2.2 \u63d2\u66f2\uff1agithub\u914d\u7f6essh-key<\/h3>\n

git submodule update \u8fd9\u4e00\u6b65\u679c\u7136\u51fa\u9519\u3002\u5c06ssh\u5730\u5740\u6539\u6210https\u5730\u5740\u597d\u51e0\u6b21\u90fd\u4e0d\u6210\u529f\uff0c\u5e72\u8106\u914d\u7f6e\u4e2assh-key\u5427\uff0c\u53cd\u6b63\u7533\u8bf7\u6d59\u5927\u670d\u52a1\u5668\u7684\u65f6\u5019\u4e5f\u7528\u7684\u8fd9\u5957\u65b9\u5f0f\u3002\u5177\u4f53\u6d41\u7a0b\u5982\u4e0b\u3002<\/strong><\/h3>\n
\n
    \n
  1. \u6253\u5f00Git Bash\u547d\u4ee4\u7a97\u53e3\u3002<\/li>\n
  2. \u8f93\u5165\u4ee5\u4e0b\u547d\u4ee4\u4ee5\u751f\u6210SSH\u5bc6\u94a5\uff1assh-keygen -t rsa -C “your_email@example.com”<\/strong>\u3002\u5176\u4e2d\uff0cyour_email@example.com<\/strong>\u662f\u60a8\u7684\u7535\u5b50\u90ae\u4ef6\u5730\u5740\u3002<\/li>\n
  3. \u6309Enter\u952e\uff0c\u63a5\u53d7\u9ed8\u8ba4\u6587\u4ef6\u4f4d\u7f6e\u548c\u6587\u4ef6\u540d\u3002<\/li>\n
  4. \u5728\u63d0\u793a\u7b26\u4e0b\u8f93\u5165\u5bc6\u7801\u77ed\u8bed\u3002\u5982\u679c\u60a8\u4e0d\u60f3\u5728\u6bcf\u6b21\u4f7f\u7528SSH\u5bc6\u94a5\u65f6\u8f93\u5165\u5bc6\u7801\u77ed\u8bed\uff0c\u8bf7\u5c06\u5176\u7559\u7a7a\u5e76\u76f4\u63a5\u6309Enter\u952e\u3002<\/li>\n
  5. \u91cd\u65b0\u8f93\u5165\u5bc6\u7801\u77ed\u8bed\u4ee5\u786e\u8ba4\u3002<\/li>\n
  6. \u5728Git Bash\u547d\u4ee4\u7a97\u53e3\u4e2d\uff0c\u8f93\u5165\u4ee5\u4e0b\u547d\u4ee4\u4ee5\u5c06SSH\u5bc6\u94a5\u6dfb\u52a0\u5230ssh-agent\uff1aeval $(ssh-agent -s)<\/strong>\u3002<\/li>\n
  7. \u8f93\u5165\u4ee5\u4e0b\u547d\u4ee4\u4ee5\u5c06SSH\u5bc6\u94a5\u6dfb\u52a0\u5230ssh-agent\uff1assh-add ~\/.ssh\/id_rsa<\/strong>\u3002\u5176\u4e2d\uff0cid_rsa<\/strong>\u662f\u60a8\u7684\u79c1\u94a5\u6587\u4ef6\u540d\u3002<\/li>\n
  8. \u5c06\u516c\u94a5\u6dfb\u52a0\u5230\u60a8\u7684GitHub\u5e10\u6237\u6216\u5176\u4ed6Git\u6258\u7ba1\u670d\u52a1\u4e2d\u3002\u590d\u5236\u516c\u94a5\u7684\u5185\u5bb9\uff0c\u7136\u540e\u5728GitHub\u6216\u5176\u4ed6Git\u6258\u7ba1\u670d\u52a1\u7684\u7f51\u7ad9\u4e0a\u6253\u5f00\u8bbe\u7f6e\u9875\u9762\uff0c\u627e\u5230SSH\u5bc6\u94a5\u90e8\u5206\uff0c\u5e76\u5c06\u516c\u94a5\u7c98\u8d34\u5230\u8be5\u90e8\u5206\u4e2d\u3002<\/li>\n<\/ol>\n<\/blockquote>\n

    \u914d\u7f6e\u4e86ssh key\u4ee5\u540e\u7ec8\u4e8egit update\u6210\u529f\u3002<\/p>\n

    2.3 \u4f7f\u7528miniconda\u5b89\u88c5\u6240\u6709\u4f9d\u8d56<\/h3>\n

    \u7136\u540e\u8fd0\u884cpip\u547d\u4ee4\u5931\u8d25\uff0capt\u5b89\u88c5python\u5931\u8d25\uff0capt\u66f4\u65b0\u5931\u8d25\uff0c\u56e0\u4e3a\u6ca1\u6709\u6743\u9650
    \n\u5f97\uff0c\u90a3\u5c31\u5b89\u88c5miniconda\u7136\u540e\u628a\u76f8\u5173\u4f9d\u8d56\u4e00\u5757\u513f\u88c5\u4e86\u3002\u5148\u770b\u592a\u4e59\u6240\u5728\u5f97\u5c01\u795e\u699c\u6846\u67b6\u5bf9\u5e94\u7684python\u7248\u672c\u5927\u6982\u662f3.10\u4ee5\u4e0a\uff0c\u56e0\u6b64\u53bb\u5b98\u7f51\u9009\u62e9\u7684\u5bf9\u5e94\u7248\u672c\u7684\u5b89\u88c5\u5305\uff1a\uff08    https:\/\/docs.conda.io\/en\/latest\/miniconda.html<\/a>\uff09
    \n\u6267\u884c\u547d\u4ee4\u4e0b\u8f7d\u5b89\u88c5\u5305\uff1a<\/p>\n

    wget https:\/\/repo.anaconda.com\/miniconda\/Miniconda3-py310_23.3.1-0-Linux-x86_64.sh<\/a><\/p><\/blockquote>\n

    \u8fd0\u884csh\u811a\u672c\u5b89\u88c5miniconda:<\/p>\n

    bash Miniconda3-py310_23.3.1-0-Linux-x86_64.sh<\/p><\/blockquote>\n

    \u63a5\u4e0b\u6765\u6309\u7167\u5b89\u88c5\u7a0b\u5e8f\u7684\u63d0\u793a\u64cd\u4f5c\uff0c\u4e00\u8def\u8f93\u5165\u56de\u8f66\u6216yes\u6765\u5b8c\u6210\u5b89\u88c5\u3002
    \n\u7136\u540e\u5173\u95ed\u5f53\u524d\u7ec8\u7aef\uff0c\u91cd\u65b0\u6253\u5f00\u4e00\u4e2a\u65b0\u7ec8\u7aef\uff0cminiconda\u624d\u80fd\u751f\u6548\u3002
    \n\u6b64\u65f6\u53ef\u4ee5\u67e5\u770b\u7248\u672c\u53f7\u547d\u4ee4\u6765\u67e5\u770bminiconda\u662f\u5426\u5df2\u7ecf\u6b63\u786e\u5b89\u88c5\u3002<\/p>\n

    conda –version
    \npython –version<\/p><\/blockquote>\n

    \u5982\u679c\u4ee5\u4e0a\u547d\u4ee4\u6267\u884c\u65e0\u8bef\uff0c\u5219\u53ef\u4ee5\u5220\u9664\u4e4b\u524d\u7684miniconda\u5b89\u88c5\u811a\u672c<\/p>\n

    rm Miniconda3-py310_23.3.1-0-Linux-x86_64.sh<\/p><\/blockquote>\n

    \u66f4\u65b0miniconda<\/p>\n

    conda install –channel defaults conda python=3.10 –yes
    \nconda update –channel defaults –all –yes<\/p><\/blockquote>\n

    \u6700\u540e\u5728Fenshenbang-LM\u6846\u67b6\u4e0b\u8fd0\u884c\u547d\u4ee4\u5b89\u88c5\u5404\u79cd\u4f9d\u8d56\uff1a
    \npip install –editable .
    \n\u4e2d\u95f4\u4e0b\u8f7dtorch\u56e0\u4e3aread time out\u4e2d\u65ad\u8fc7\u4e00\u6b21\uff0c\u4e0d\u8fc7\u7b2c\u4e8c\u5929\u91cd\u65b0\u6267\u884c\u4e0a\u9762\u7684\u547d\u4ee4\u540e\u6210\u529f\u5b89\u88c5\u3002
    \n
    \n\u53ef\u4ee5\u770b\u5230\u4f9d\u8d56\u975e\u5e38\u7684\u591a\u3002\u5176\u4e2d\u5173\u952e\u7684\u662ftorch\u7248\u672c\u662f2.0.1, \u8fd9\u4e2a\u5f88\u6709\u53ef\u80fd\u4f1a\u5bfc\u81f4\u62a5\u9519\uff0c\u56e0\u4e3a\u6211\u770b\u5230\u5728\u5b98\u65b9\u6587\u6863\u4e2d\u7684docker\u5b89\u88c5\u8bf4\u660e\u4e2d\uff0c\u4f7f\u7528\u7684pytorch\u7248\u672c\u662f1.10\u3002\u800c\u4f17\u6240\u5468\u77e5\u4ecepytorch1.X\u52302.X\u662f\u4e00\u6b21\u91cd\u5927\u7248\u672c\u66f4\u65b0\uff0c\u5f88\u591abug\u90fd\u6ca1\u4fee\uff0c\u5b58\u5728\u4e0d\u80fd\u5411\u4e0b\u517c\u5bb9\u7684\u95ee\u9898\u3002\u6240\u4ee5\u7279\u522b\u8bb0\u5f55\u4e00\u4e0b\uff08F**K\u540e\u9762\u679c\u7136\u56e0\u4e3a\u8fd9\u4e2a\u62a5\u9519\u4e86\uff09\u3002<\/p>\n

    \n

    \u7b2c\u4e09\u6b65\uff1a\u8fd0\u884c\u592a\u4e59\u6a21\u578b\u7684finetune example<\/h2>\n

    \u5148\u6253\u5f00Fengshenbang-LM<\/a>\/fengshen<\/a>\/examples<\/a>\/finetune_taiyi_stable_diffusion<\/strong>\/\u76ee\u5f55\u4e0b\u7684finetune.sh\u67e5\u770b\u7ec6\u8282\u3002
    \n\u5982\u4e0a\u56fe\u6240\u793a\uff0c\u6587\u6863\u9ed8\u8ba4\u662f\u91c7\u7528deepspeed\u7684\u534a\u7cbe\u5ea6\u8bad\u7ec3\u3002\u7406\u8bba\u4e0a\u4e00\u5f203090\u663e\u5361\u5c31\u80fd\u505a\u3002
    \n\u6267\u884c\u4e00\u6b21finetune.sh\u811a\u672c\u8bd5\u8bd5
    \nbash finetune.sh
    \n\u679c\u7136\u62a5\u9519\uff0c\u8bf4\u6ca1\u6709\u53eb\u505adiffusers\u7684module
    \n\u90a3\u5c31\u8fd0\u884cpip install diffusers\u5b89\u88c5\u4e00\u4e0b\u8fd9\u4e2a\u5e93
    \n\u4f9d\u6b64\u7c7b\u63a8\uff0c\u4f46\u51e1\u9047\u5230\u7f3a\u5e93\u7684\u95ee\u9898\u90fd\u8fd9\u4e48\u89e3\u51b3
    \n\u7136\u540e\u518d\u6267\u884cfinetune.sh\u6587\u4ef6\uff0c\u53c8\u62a5\u9519\uff1a
    \n
    \n\u8bf4Trainer\u6ca1\u6709\u540d\u4e3aadd_argparse_args\u7684\u5c5e\u6027\uff0c\u8fd9\u4e0d\u662f\u7f16\u7a0b\u624d\u6709\u7684\u9519\u8bef\u5417\uff1f\u95ee\u9898\u662f\u8fd9\u7a0b\u5e8f\u662f\u5b98\u65b9\u6587\u6863\u7ed9\u7684\uff0c\u6211\u8fd8\u6ca1\u6765\u5f97\u53ca\u6539\u4efb\u4f55\u4e1c\u897f\u5462\u3002\u770b\u770b\u6700\u7ec8\u7684\u6267\u884c\u6587\u4ef6finetune.py\u6392\u9519\uff1a
    \n
    \n\u8fd9Trainer\u5bf9\u8c61\u6765\u81ea\u4e8epytorch_lightning\uff0c\u8fd9\u662f\u4e00\u4e2a\u8f7b\u91cf\u7ea7\u7684pytorch\u5e93\uff0c\u6216\u8005\u8bf4\u672c\u9879\u76ee\u7684torch\u5bf9\u8c61\u90fd\u662f\u4f9d\u8d56\u4e8epytorch_lighting\u7684\uff0c\u8fd9\u4e2a\u5e93\u7f3a\u5bf9\u8c61\u4e86\u600e\u4e48\u53ef\u80fd\uff1f\u8fd9\u56de\u7ed9\u6211\u6574\u4e0d\u4f1a\u4e86<\/p>\n

    2.4 \u63d2\u66f2\uff1apytorch\u7248\u672c\u5bfc\u81f4\u7684bug\u4fee\u590d<\/h3>\n

    \u4e0d\u884c\u4e86\uff0c\u642c\u6551\u5175\uff1a
    \n
    \n\u5f20\u5efa\u4f1f\u535a\u58eb\u51ed\u501f\u7ecf\u9a8c\u5224\u65ad\u5e94\u8be5\u662f\u7248\u672c\u95ee\u9898\u3002
    \n\u9996\u5148\u67e5\u8be2pytorch\u7248\u672c
    \nconda list|grep pytorch
    \n\u7136\u540e\u67e5\u8be2\u8be5\u7248\u672c\u7684pytorch\u7684Trainer\uff0c\u6709\u6ca1\u6709\u540d\u4e3aadd_argparse_args\u7684\u5c5e\u6027\u3002\u7ed3\u8bba\u662f\u6ca1\u6709\u3002
    \n\u53c8\u53bb    https:\/\/lightning.ai\/docs\/pytorch\/2.0.2\/search.html<\/a> \u5b98\u65b9\u6587\u6863\u91cc\u67e5\u4e86\u4e00\u4e0b\uff0c\u786e\u5b9a\u662f\u7248\u672c\u7684\u95ee\u9898\u5bfc\u81f4\u6ca1\u6709\u5bf9\u5e94\u7684\u5c5e\u6027\u548c\u65b9\u6cd5
    \n\u4f7f\u7528vscode\u8fdc\u7a0b\u8fde\u63a5\u670d\u52a1\u5668\uff0c\u4fee\u6539fengshen\u7684\u914d\u7f6e\u5b89\u88c5\u6587\u4ef6setup.py\uff0c \u7ed9pytorch_lighting\u7684\u5b89\u88c5\u7248\u672c\u589e\u52a0\u5224\u65ad\u6761\u4ef6<2\uff0c\u8ba9\u5b83\u4e0d\u8981\u5b89\u88c52.0\u4ee5\u4e0a\u7248\u672c
    \n
    \n\u7136\u540e\u91cd\u65b0\u8fd0\u884csetup.py\u6587\u4ef6
    \npip install –editable .
    \n\u4e8e\u662f\u7cfb\u7edf\u91cd\u88c5\u4e861.9.5\u7248\u672c\u7684pytorch_lighting
    \n\u5f20\u5efa\u4f1f\u535a\u58eb\u529f\u6210\u8eab\u9000\u3002\u5bf9\u4e8e\u6211\u8fd9\u79cd\u6ca1\u6709\u8bad\u7ec3\u6a21\u578b\u7ecf\u9a8c\u7684\u4eba\uff0c\u9047\u5230\u8fd9\u79cd\u95ee\u9898\uff0c\u6ca1\u6709\u4ed9\u4eba\u6307\u8def\u4f30\u8ba1\u5f97\u6574\u597d\u51e0\u5929\u8017\u5c3d\u8010\u5fc3\u3002
    \n\u518d\u6267\u884cbash finetune.sh
    \n\u679c\u7136\u6ca1\u6709\u4e4b\u524d\u7684\u9519\u8bef\u4e86\uff08\u8be5\u4e0b\u4e00\u4e2a\u9519\u4e86\uff09<\/p>\n

    \u63a5\u4e0b\u6765\u7684\u9519\u8bef\u662f\u8fd0\u884cfinetune.sh\u65f6\uff0c\u59cb\u7ec8\u65e0\u6cd5\u987a\u5229\u7684\u4ecehuggingface\u4e0a\u4e0b\u8f7d\u6a21\u578b<\/strong><\/p>\n

    \n

    2.5 \u4e0b\u8f7d\u548c\u4f7f\u7528\u6a21\u578b<\/h3>\n

    \u6709\u4e24\u6761\u8def\u7ebf\uff1a<\/p>\n

      \n
    1. \u76f4\u63a5\u8fd0\u884cbash finetune.sh\uff0c\u4f1a\u4f9d\u636e\u811a\u672c\u81ea\u52a8\u5730\u4ecehuggingface\u4e0a\u4e0b\u8f7d\u6a21\u578b<\/li>\n
    2. \u81ea\u5df1\u5148\u4ecehuggingface\u4e0aclone\u6a21\u578b\u5230\u672c\u5730\uff0c\u7136\u540e\u4fee\u6539finetune.sh\u4e2d\u7684model-path\uff0c\u4ece\u800c\u5b9e\u73b0\u8ddf1\u7c7b\u4f3c\u7684\u76ee\u6807<\/li>\n<\/ol>\n

      \u4e24\u6761\u8def\u7ebf\u6211\u90fd\u8bd5\u4e86\uff0c\u90fd\u5931\u8d25\u4e86\u597d\u591a\u6b21\u3002\u5177\u4f53\u5982\u4e0b\uff1a<\/p>\n

      2.5.1 \u76f4\u63a5\u4ecehuggingface\u4e0a\u4e0b\u8f7d\u6a21\u578b<\/h4>\n

      \u76f4\u63a5\u8fd0\u884cbash finetune.sh\uff0c\u8bd5\u4e865\uff0c6\u6b21\uff0c\u6bcf\u6b21\u90fd\u770b\u8d77\u6765\u5f88\u6709\u5e0c\u671b\u5730\u4e0b\u8f7d\u6587\u4ef6\uff08\u603b\u517114\u4e2a\uff09\uff0c\u7136\u540e\u4e0b\u5230\u4e00\u534a\u7684\u65f6\u5019read time out\uff0c\u7b80\u5355\u7684\u8bf4\u5c31\u662f\u7f51\u7edc\u8fde\u63a5\u8d85\u65f6\/\u8d85\u8fc7\u4e86\u5c1d\u8bd5\u6b21\u6570\uff0c\u65ad\u5f00\u4e86\u8fde\u63a5<\/p>\n

      \u7136\u540e\u505a\u4e86\u6d4b\u8bd5\u7f51\u7edc\u901f\u5ea6\u7b49\u4e00\u7cfb\u5217\u7684\u5de5\u4f5c\uff0c\u65e0\u8bba\u662f\u4f7f\u7528Ping\u547d\u4ee4\uff0ccurl -I \u547d\u4ee4\uff0c\u8fd8\u662fiperf\u4e4b\u7c7b\u7684\u5de5\u5177\uff0c\u90fd\u6ca1\u53d1\u73b0\u6709\u95ee\u9898\u3002\u7f51\u901f\u90fd\u662f\u6d59\u5927\u8fd9\u8fb9\u6807\u51c6\u76843.5MB\u6700\u9ad8\u7684\u7f51\u901f\u3002
      \n\u4e8e\u662f\u60f3\u627e\u4e2ahuggingface\u7684\u955c\u50cf\uff0c\u7ed3\u679c\uff1a
      \n
      \n\u8fd8\u6709\u4e00\u4e2a\u529e\u6cd5\u662f\u4fee\u6539\u7f51\u7edc\u8fde\u63a5\u7684timeout\u503c\uff0c\u6bd4\u5982\u4ece10\u6539\u523030\uff0c\u5e94\u8be5\u80fd\u7f13\u89e3\u8fd9\u4e2a\u95ee\u9898\uff0c\u4f46\u6682\u65f6\u8fd8\u4e0d\u6e05\u695a\u600e\u4e48\u53bb\u4fee\u6539\u3002
      \n\u7136\u540e\u8fd9\u4e2a\u4e0b\u8f7d\u4e5f\u6bd4\u8f83\u5947\u602a\u7684\u4e00\u70b9\u662f\uff0c\u6bcf\u6b21\u91cd\u65b0\u4e0b\u8f7d\u90fd\u662f\u8981\u628a\u6240\u6709\u7684\u4e1c\u897f\u5168\u90e8\u91cd\u65b0\u5f00\u59cb\uff0c\u800c\u4e0d\u662f\u4ece\u4e0a\u6b21\u672a\u5b8c\u6210\u7684\u7ec4\u4ef6\u5f00\u662f\u4e0b\u8f7d\u3002\u4e0b\u8f7d\u7684\u6587\u4ef6\u4e5f\u4e0d\u77e5\u9053\u653e\u5728\u90a3\u91cc\u2026\u2026 \u54e6\uff0c\u662f\u5728cache\u91cc
      \n
      \n\u8bf4\u5b9e\u8bdd\u6211\u4e5f\u641e\u4e0d\u6e05\u8fd9\u4e9b\u540d\u5b57\u5947\u602a\u7684\u6587\u4ef6\u54ea\u4e2a\u662f\u54ea\u4e2a\uff0c\u53ea\u77e5\u9053\u4e0d\u90fd\u4e0b\u597d\u4e86\u5c31\u4e0d\u80fd\u7528\u3002\u9042\u6539\u4e3a\u7b2c\u4e8c\u6761\u8def\u7ebf\uff1a<\/p>\n

      2.5.1 \u76f4\u63a5\u4e0b\u8f7d\u6a21\u578b\u540e\u4fee\u6539model_path<\/h4>\n

      \u9996\u5148\u6309\u7167\u59dc\u667a\u5fb7\u4e4b\u524d\u7684\u6587\u6863\uff0c\u7528git\u4e0b\u8f7d\u6a21\u578b\u7136\u540e\u4fee\u6539model_path
      \n\u5148\u5230workspace\u76ee\u5f55\u4f7f\u7528git\u4e0b\u8f7d\u6570\u636e
      \ncd ~\/Fengshenbang-LM\/fengshen\/workspace\/Taiyi-Stable-Diffusion-1B-Chinese-v0.1
      \ngit clone [https:\/\/huggingface.co\/IDEA-CCNL\/Taiyi-Stable-Diffusion-1B-Chinese-v0.1](https:\/\/huggingface.co\/IDEA-CCNL\/Taiyi-Stable-Diffusion-1B-Chinese-v0.1)<\/code>
      \n\u7136\u540e\u4fee\u6539example \u76ee\u5f55\u4e0b\u7684 finetune.sh\uff0c\u5c06model_path\u6539\u6210\u4e0b\u8f7d\u8def\u5f84\u7684workspace\/Taiyi-Stable-Diffusion-1B-Chinese-v0.1\uff0c
      \n\u7136\u540e\u6267\u884cbash finetune.sh, \u62a5\u9519
      \n\u7b80\u5355\u7684\u8bf4\uff0c\u5c31\u662f\u7cfb\u7edf\u6ca1\u6709\u627e\u5230\u6211\u4e0b\u8f7d\u7684\u6a21\u578b\uff0c\u9ed8\u8ba4\u8fd8\u662f\u5f97\u4ecehuggingface\u4e0a\u4e0b\u8f7d\uff0c\u800c\u4e14\u8fd9\u4e2a\u8def\u5f84\u8fd8\u6709\u95ee\u9898\uff0c\u62a5401\u6ca1\u6709\u6743\u9650\u4e0b\u8f7d\u6216404\u627e\u4e0d\u5230\u5bf9\u5e94\u6587\u4ef6<\/p>\n

      \u9042\u5c06\u8def\u5f84\u6539\u4e3a\u7edd\u5bf9\u8def\u5f84\uff1a<\/strong>
      \n
      \n\u518d\u6267\u884cbash finetune.sh, \u6210\u529f\u907f\u514d\u4e86\u8def\u5f84\u7684\u9519\u8bef\uff0c\u6267\u884c\u5230\u4e0b\u4e00\u4e2a\u9519\u8bef\u4e86\u3002<\/p>\n

      2.5.3 \u5207\u6362\u6e05\u534e\u955c\u50cf<\/h4>\n

      \u4e0b\u4e00\u4e2a\u9519\u8bef\u662f\u7cfb\u7edf\u63d0\u793a\u7f3a\u5c11xformers\u8fd9\u4e2amodule
      \n\u4f7f\u7528pip install xformers\u8fde\u7eed\u7f51\u7edc\u8d85\u65f6
      \n\u95ee\u6d59\u5927\u670d\u52a1\u5668\u7ba1\u7406\u5458\u6d59\u5927\u535a\u58eb\u51af\u6d69\u54f2\uff0c\u7b54\u66f0\u670d\u52a1\u5668\u4e0d\u63d0\u4f9b\u79d1\u5b66\u4e0a\u7f51\uff0c\u8fd9\u79cd\u60c5\u51b5\u8bf7\u4f7f\u7528\u6e05\u534etuna\u955c\u50cf\uff0c\u8be6\u89c1
      \n      https:\/\/mirrors.tuna.tsinghua.edu.cn\/help\/pypi\/<\/a>
      \n\u603b\u800c\u8a00\u4e4b\u4f7f\u7528\u6e05\u534etuna\u955c\u50cf\u4ee5\u540e\u679c\u7136\u4e0b\u8f7d\u795e\u901f\uff0cpip install xformers\u5b8c\u6210<\/p>\n

      2.6 fuse_adam\u9519\u8bef\u7684\u89e3\u51b3\u4e0e\u8bad\u7ec3\u6210\u529f<\/h3>\n

      \u7136\u540e\u4e0b\u4e00\u4e2a\u9519\u8bef\uff0c\u8bf4RuntimeError: Error building extension ‘fused_adam’
      \n\u67e5\u4e86\u4e00\u4e0b\uff0c”fused_adam” \u662f\u4e00\u4e2a\u7528\u4e8e\u6df1\u5ea6\u5b66\u4e60\u6846\u67b6 PyTorch \u7684\u6269\u5c55\uff0c\u5b83\u63d0\u4f9b\u4e86\u4f18\u5316\u7684 Adam \u4f18\u5316\u5668\u5b9e\u73b0\u3002Adam \u662f\u4e00\u79cd\u5e38\u7528\u7684\u4f18\u5316\u7b97\u6cd5\uff0c\u7528\u4e8e\u8bad\u7ec3\u795e\u7ecf\u7f51\u7edc\u6a21\u578b\u3002”fused_adam” \u7684\u8bbe\u8ba1\u76ee\u7684\u662f\u901a\u8fc7\u878d\u5408\u591a\u4e2a\u64cd\u4f5c\uff0c\u63d0\u9ad8 Adam \u4f18\u5316\u5668\u7684\u8ba1\u7b97\u6548\u7387\u3002\u4e0e\u5fae\u8f6f\u7684deepSpeed\u6709\u5f88\u6df1\u7684\u5173\u7cfb\u3002<\/p>\n

      2.6.1 \u6b7b\u78d5fuse_adam\u9519\u8bef<\/h4>\n

      chatGPT\u7684\u56de\u7b54\uff1a\u5f53\u60a8\u9047\u5230 “RuntimeError: Error building extension ‘fused_adam'” \u9519\u8bef\u65f6\uff0c\u901a\u5e38\u8868\u793a\u5728\u6784\u5efa\u6216\u5b89\u88c5 “fused_adam” \u6269\u5c55\u65f6\u51fa\u73b0\u4e86\u95ee\u9898\u3002\u53ef\u80fd\u7684\u539f\u56e0\u5305\u62ec\uff1a<\/p>\n