{"id":28303,"date":"2025-03-12T10:33:24","date_gmt":"2025-03-12T02:33:24","guid":{"rendered":"https:\/\/www.aisharenet.com\/?p=28303"},"modified":"2025-03-12T10:33:24","modified_gmt":"2025-03-12T02:33:24","slug":"ai-toolkit-by-ostris","status":"publish","type":"post","link":"https:\/\/www.kdjingpai.com\/pt\/ai-toolkit-by-ostris\/","title":{"rendered":"AI Toolkit by Ostris\uff1aStable Diffusion\u4e0eFLUX.1\u6a21\u578b\u8bad\u7ec3\u5de5\u5177\u5305"},"content":{"rendered":"<p>AI Toolkit by Ostris \u662f\u4e00\u4e2a\u5f00\u6e90\u7684AI\u5de5\u5177\u96c6\uff0c\u4e13\u6ce8\u4e8e\u652f\u6301Stable Diffusion\u53caFLUX.1\u6a21\u578b\u7684\u8bad\u7ec3\u4e0e\u56fe\u50cf\u751f\u6210\u4efb\u52a1\u3002\u8be5\u5de5\u5177\u96c6\u7531\u5f00\u53d1\u8005Ostris\u521b\u5efa\u5e76\u7ef4\u62a4\uff0c\u6258\u7ba1\u4e8eGitHub\uff0c\u65e8\u5728\u4e3a\u7814\u7a76\u8005\u548c\u5f00\u53d1\u8005\u63d0\u4f9b\u7075\u6d3b\u7684\u6a21\u578b\u5fae\u8c03\u4e0e\u5b9e\u9a8c\u5e73\u53f0\u3002\u5b83\u5305\u542b\u591a\u79cdAI\u811a\u672c\uff0c\u652f\u6301LoRA\u63d0\u53d6\u3001\u56fe\u50cf\u6279\u91cf\u751f\u6210\u53ca\u7279\u5b9a\u5c42\u8bad\u7ec3\u7b49\u529f\u80fd\u3002\u76ee\u524d\u9879\u76ee\u5904\u4e8e\u5f00\u53d1\u9636\u6bb5\uff0c\u90e8\u5206\u529f\u80fd\u53ef\u80fd\u4e0d\u591f\u7a33\u5b9a\uff0c\u4f46\u5176\u9ad8\u5ea6\u53ef\u5b9a\u5236\u6027\u4f7f\u5176\u9002\u7528\u4e8e\u6df1\u5ea6\u5b66\u4e60\u9886\u57df\u7684\u9ad8\u7ea7\u7528\u6237\u3002\u5de5\u5177\u96c6\u652f\u6301Linux\u548cWindows\u7cfb\u7edf\uff0c\u8fd0\u884cFLUX.1\u6a21\u578b\u8bad\u7ec3\u9700\u81f3\u5c1124GB\u663e\u5b58\u7684Nvidia GPU\u3002<\/p>\n<p><img loading=\"lazy\" decoding=\"async\" class=\"aligncenter size-full wp-image-28304\" title=\"AI Toolkit by Ostris\uff1aStable Diffusion\u4e0eFLUX.1\u6a21\u578b\u8bad\u7ec3\u5de5\u5177\u5305-1\" src=\"https:\/\/www.kdjingpai.com\/wp-content\/uploads\/2025\/03\/bbb33fa148cbc6b.jpg\" alt=\"AI Toolkit by Ostris\uff1aStable Diffusion\u4e0eFLUX.1\u6a21\u578b\u8bad\u7ec3\u5de5\u5177\u5305-1\" width=\"1309\" height=\"512\" srcset=\"https:\/\/www.kdjingpai.com\/wp-content\/uploads\/2025\/03\/bbb33fa148cbc6b.jpg 1309w, https:\/\/www.kdjingpai.com\/wp-content\/uploads\/2025\/03\/bbb33fa148cbc6b-768x300.jpg 768w, https:\/\/www.kdjingpai.com\/wp-content\/uploads\/2025\/03\/bbb33fa148cbc6b-18x7.jpg 18w\" sizes=\"auto, (max-width: 1309px) 100vw, 1309px\" \/><\/p>\n<p>&nbsp;<\/p>\n<h2>\u529f\u80fd\u5217\u8868<\/h2>\n<ul>\n<li><strong>\u6a21\u578b\u8bad\u7ec3<\/strong>: \u652f\u6301Stable Diffusion\u53caFLUX.1\u6a21\u578b\u5fae\u8c03\uff0c\u53ef\u8bad\u7ec3LoRA\u548cLoKr\u6a21\u578b\u3002<\/li>\n<li><strong>\u56fe\u50cf\u751f\u6210<\/strong>: \u6839\u636e\u914d\u7f6e\u6587\u4ef6\u6216\u6587\u672c\u63d0\u793a\u751f\u6210\u6279\u91cf\u56fe\u50cf\u3002<\/li>\n<li><strong>LoRA\u63d0\u53d6\u4e0e\u4f18\u5316<\/strong>: \u63d0\u4f9bLoRA\u548cLoCON\u63d0\u53d6\u5de5\u5177\uff0c\u4f18\u5316\u6a21\u578b\u7279\u5f81\u63d0\u53d6\u3002<\/li>\n<li><strong>\u7279\u5b9a\u5c42\u8bad\u7ec3<\/strong>: \u53ef\u9009\u62e9\u7279\u5b9a\u795e\u7ecf\u7f51\u7edc\u5c42\u8fdb\u884c\u8bad\u7ec3\uff0c\u7075\u6d3b\u8c03\u6574\u6743\u91cd\u3002<\/li>\n<li><strong>\u7528\u6237\u754c\u9762\u652f\u6301<\/strong>: \u63d0\u4f9bAI Toolkit UI\u548cGradio UI\uff0c\u7b80\u5316\u4efb\u52a1\u7ba1\u7406\u548c\u6a21\u578b\u8bad\u7ec3\u64cd\u4f5c\u3002<\/li>\n<li><strong>\u6570\u636e\u96c6\u5904\u7406<\/strong>: \u81ea\u52a8\u8c03\u6574\u56fe\u50cf\u5206\u8fa8\u7387\u5e76\u6309\u6876\u5206\u7ec4\uff0c\u652f\u6301\u591a\u79cd\u56fe\u50cf\u683c\u5f0f\u3002<\/li>\n<li><strong>\u4e91\u7aef\u8bad\u7ec3<\/strong>: \u652f\u6301\u5728RunPod\u548cModal\u5e73\u53f0\u8fd0\u884c\u8bad\u7ec3\u4efb\u52a1\u3002<\/li>\n<\/ul>\n<p>&nbsp;<\/p>\n<h2>\u4f7f\u7528\u5e2e\u52a9<\/h2>\n<h3>\u5b89\u88c5\u6d41\u7a0b<\/h3>\n<h4>Linux\u7cfb\u7edf\u5b89\u88c5<\/h4>\n<ol>\n<li><strong>\u514b\u9686\u4ed3\u5e93<\/strong>: \u5728\u7ec8\u7aef\u8fd0\u884c\u4ee5\u4e0b\u547d\u4ee4\u4e0b\u8f7d\u4ee3\u7801\uff1a<\/li>\n<\/ol>\n<pre><code>git clone https:\/\/github.com\/ostris\/ai-toolkit.git\r\ncd ai-toolkit\r\n<\/code><\/pre>\n<ol start=\"2\">\n<li><strong>\u66f4\u65b0\u5b50\u6a21\u5757<\/strong>: \u786e\u4fdd\u6240\u6709\u4f9d\u8d56\u5e93\u5b8c\u6574\uff1a<\/li>\n<\/ol>\n<pre><code>git submodule update --init --recursive\r\n<\/code><\/pre>\n<ol start=\"3\">\n<li><strong>\u521b\u5efa\u865a\u62df\u73af\u5883<\/strong>: \u4f7f\u7528Python 3.10\u6216\u66f4\u9ad8\u7248\u672c\uff1a<\/li>\n<\/ol>\n<pre><code>python3 -m venv venv\r\nsource venv\/bin\/activate\r\n<\/code><\/pre>\n<ol start=\"4\">\n<li><strong>\u5b89\u88c5\u4f9d\u8d56<\/strong>: \u5148\u5b89\u88c5PyTorch\uff0c\u518d\u5b89\u88c5\u5176\u4ed6\u4f9d\u8d56\uff1a<\/li>\n<\/ol>\n<pre><code>pip3 install torch\r\npip3 install -r requirements.txt\r\n<\/code><\/pre>\n<h4>Windows\u7cfb\u7edf\u5b89\u88c5<\/h4>\n<ol>\n<li><strong>\u514b\u9686\u4ed3\u5e93<\/strong>: \u5728\u547d\u4ee4\u63d0\u793a\u7b26\u4e2d\u8fd0\u884c\uff1a<\/li>\n<\/ol>\n<pre><code>git clone https:\/\/github.com\/ostris\/ai-toolkit.git\r\ncd ai-toolkit\r\n<\/code><\/pre>\n<ol start=\"2\">\n<li><strong>\u66f4\u65b0\u5b50\u6a21\u5757<\/strong>:<\/li>\n<\/ol>\n<pre><code>git submodule update --init --recursive\r\n<\/code><\/pre>\n<ol start=\"3\">\n<li><strong>\u521b\u5efa\u865a\u62df\u73af\u5883<\/strong>:<\/li>\n<\/ol>\n<pre><code>python -m venv venv\r\n.\\venv\\Scripts\\activate\r\n<\/code><\/pre>\n<ol start=\"4\">\n<li><strong>\u5b89\u88c5\u4f9d\u8d56<\/strong>: \u5b89\u88c5\u652f\u6301CUDA 12.4\u7684PyTorch\u7248\u672c\uff0c\u518d\u5b89\u88c5\u5176\u4ed6\u4f9d\u8d56\uff1a<\/li>\n<\/ol>\n<pre><code>pip install torch==2.5.1 torchvision==0.20.1 --index-url https:\/\/download.pytorch.org\/whl\/cu124\r\npip install -r requirements.txt\r\n<\/code><\/pre>\n<h4>UI\u754c\u9762\u5b89\u88c5<\/h4>\n<ol>\n<li><strong>\u5b89\u88c5Node.js<\/strong>: \u786e\u4fdd\u7cfb\u7edf\u5b89\u88c5Node.js 18\u6216\u66f4\u9ad8\u7248\u672c\u3002<\/li>\n<li><strong>\u6784\u5efaUI<\/strong>: \u8fdb\u5165ui\u76ee\u5f55\u5e76\u5b89\u88c5\u4f9d\u8d56\uff1a<\/li>\n<\/ol>\n<pre><code>cd ui\r\nnpm install\r\nnpm run build\r\nnpm run update_db\r\n<\/code><\/pre>\n<ol start=\"3\">\n<li><strong>\u8fd0\u884cUI<\/strong>: \u542f\u52a8\u754c\u9762\uff1a<\/li>\n<\/ol>\n<pre><code>npm run start\r\n<\/code><\/pre>\n<ol start=\"4\">\n<li><strong>\u8bbf\u95eeUI<\/strong>: \u5728\u6d4f\u89c8\u5668\u4e2d\u8f93\u5165<code>http:\/\/localhost:8675<\/code>\u3002<\/li>\n<\/ol>\n<h3>\u4e3b\u8981\u529f\u80fd\u64cd\u4f5c\u6d41\u7a0b<\/h3>\n<h4>FLUX.1\u6a21\u578b\u8bad\u7ec3<\/h4>\n<ol>\n<li><strong>\u51c6\u5907\u73af\u5883<\/strong>: \u786e\u4fddGPU\u663e\u5b58\u81f3\u5c1124GB\u3002\u82e5\u7528\u4e8e\u663e\u793a\u8f93\u51fa\uff0c\u9700\u5728\u914d\u7f6e\u6587\u4ef6\u4e2d\u8bbe\u7f6e<code>low_vram: true<\/code>\uff0c\u4ee5\u5728CPU\u4e0a\u91cf\u5316\u6a21\u578b\u3002<\/li>\n<li><strong>\u914d\u7f6eFLUX.1-dev<\/strong>:<\/li>\n<\/ol>\n<ul>\n<li>\u767b\u5f55Hugging Face\uff0c\u8bbf\u95ee<a href=\"https:\/\/huggingface.co\/black-forest-labs\/FLUX.1-dev\">black-forest-labs\/FLUX.1-dev<\/a>\u5e76\u63a5\u53d7\u8bb8\u53ef\u3002<\/li>\n<li>\u5728\u9879\u76ee\u6839\u76ee\u5f55\u521b\u5efa<code>.env<\/code>\u6587\u4ef6\uff0c\u6dfb\u52a0<code>HF_TOKEN=\u4f60\u7684\u8bfb\u53d6\u5bc6\u94a5<\/code>\u3002<\/li>\n<\/ul>\n<ol start=\"3\">\n<li><strong>\u914d\u7f6eFLUX.1-schnell<\/strong>:<\/li>\n<\/ol>\n<ul>\n<li>\u7f16\u8f91\u914d\u7f6e\u6587\u4ef6\uff08\u5982<code>train_lora_flux_schnell_24gb.yaml<\/code>\uff09\uff0c\u6dfb\u52a0\uff1a\n<pre><code>model:\r\nname_or_path: \"black-forest-labs\/FLUX.1-schnell\"\r\nassistant_lora_path: \"ostris\/FLUX.1-schnell-training-adapter\"\r\nis_flux: true\r\nquantize: true\r\nsample:\r\nguidance_scale: 1\r\nsample_steps: 4\r\n<\/code><\/pre>\n<\/li>\n<\/ul>\n<ol start=\"4\">\n<li><strong>\u51c6\u5907\u6570\u636e\u96c6<\/strong>: \u5728\u6839\u76ee\u5f55\u521b\u5efa<code>dataset<\/code>\u6587\u4ef6\u5939\uff0c\u653e\u5165<code>.jpg<\/code>\u3001<code>.jpeg<\/code>\u6216<code>.png<\/code>\u56fe\u50cf\u53ca\u5bf9\u5e94\u7684<code>.txt<\/code>\u63cf\u8ff0\u6587\u4ef6\u3002<\/li>\n<li><strong>\u7f16\u8f91\u914d\u7f6e\u6587\u4ef6<\/strong>: \u590d\u5236<code>config\/examples\/train_lora_flux_24gb.yaml<\/code>\u5230<code>config<\/code>\u76ee\u5f55\uff0c\u91cd\u547d\u540d\u4e3a<code>my_config.yml<\/code>\uff0c\u4fee\u6539<code>folder_path<\/code>\u4e3a\u6570\u636e\u96c6\u8def\u5f84\u3002<\/li>\n<li><strong>\u8fd0\u884c\u8bad\u7ec3<\/strong>: \u6267\u884c\uff1a<\/li>\n<\/ol>\n<pre><code>python run.py config\/my_config.yml\r\n<\/code><\/pre>\n<p>\u8bad\u7ec3\u7ed3\u679c\u4fdd\u5b58\u4e8e\u6307\u5b9a\u8f93\u51fa\u6587\u4ef6\u5939\uff0c\u53ef\u7528Ctrl+C\u6682\u505c\u5e76\u4ece\u6700\u8fd1\u68c0\u67e5\u70b9\u6062\u590d\u3002<\/p>\n<h4>\u4f7f\u7528Gradio UI\u8bad\u7ec3<\/h4>\n<ol>\n<li><strong>\u767b\u5f55Hugging Face<\/strong>: \u8fd0\u884c<code>huggingface-cli login<\/code>\uff0c\u8f93\u5165\u5177\u6709<code>write<\/code>\u6743\u9650\u7684\u5bc6\u94a5\u3002<\/li>\n<li><strong>\u542f\u52a8UI<\/strong>: \u6267\u884c\uff1a<\/li>\n<\/ol>\n<pre><code>python flux_train_ui.py\r\n<\/code><\/pre>\n<ol start=\"3\">\n<li><strong>\u64cd\u4f5cUI<\/strong>: \u5728\u754c\u9762\u4e2d\u4e0a\u4f20\u56fe\u50cf\u3001\u586b\u5199\u63cf\u8ff0\u3001\u8bbe\u7f6e\u53c2\u6570\u540e\u70b9\u51fb\u8bad\u7ec3\uff0c\u5b8c\u6210\u540e\u53ef\u53d1\u5e03LoRA\u6a21\u578b\u3002<\/li>\n<\/ol>\n<h4>\u4e91\u7aef\u8bad\u7ec3\uff08RunPod\uff09<\/h4>\n<ol>\n<li><strong>\u521b\u5efaRunPod\u5b9e\u4f8b<\/strong>: \u4f7f\u7528\u6a21\u677f<code>runpod\/pytorch:2.2.0-py3.10-cuda12.1.1-devel-ubuntu22.04<\/code>\uff0c\u9009\u62e9A40\uff0848GB\u663e\u5b58\uff09\u3002<\/li>\n<li><strong>\u5b89\u88c5\u5de5\u5177\u96c6<\/strong>: \u8fde\u63a5Jupyter Notebook\uff0c\u5728\u7ec8\u7aef\u8fd0\u884cLinux\u5b89\u88c5\u547d\u4ee4\u3002<\/li>\n<li><strong>\u4e0a\u4f20\u6570\u636e\u96c6<\/strong>: \u5728\u6839\u76ee\u5f55\u521b\u5efa<code>dataset<\/code>\u6587\u4ef6\u5939\uff0c\u62d6\u5165\u56fe\u50cf\u548c\u63cf\u8ff0\u6587\u4ef6\u3002<\/li>\n<li><strong>\u914d\u7f6e\u5e76\u8fd0\u884c<\/strong>: \u4fee\u6539\u914d\u7f6e\u6587\u4ef6\u4e2d\u7684<code>folder_path<\/code>\uff0c\u6267\u884c<code>python run.py config\/my_config.yml<\/code>\u3002<\/li>\n<\/ol>\n<h4>\u6570\u636e\u96c6\u51c6\u5907<\/h4>\n<ul>\n<li><strong>\u683c\u5f0f\u8981\u6c42<\/strong>: \u652f\u6301<code>.jpg<\/code>\u3001<code>.jpeg<\/code>\u3001<code>.png<\/code>\u683c\u5f0f\uff0c\u63cf\u8ff0\u6587\u4ef6\u4e3a<code>.txt<\/code>\uff0c\u6587\u4ef6\u540d\u9700\u4e00\u81f4\uff08\u5982<code>image1.jpg<\/code>\u5bf9\u5e94<code>image1.txt<\/code>\uff09\u3002<\/li>\n<li><strong>\u63cf\u8ff0\u5185\u5bb9<\/strong>:\u00a0<code>.txt<\/code>\u4e2d\u5199\u5165\u63cf\u8ff0\uff0c\u53ef\u4f7f\u7528<code>[trigger]<\/code>\u5360\u4f4d\u7b26\uff0c\u7531\u914d\u7f6e\u6587\u4ef6\u4e2d\u7684<code>trigger_word<\/code>\u66ff\u6362\u3002<\/li>\n<li><strong>\u81ea\u52a8\u8c03\u6574<\/strong>: \u5de5\u5177\u6839\u636e\u914d\u7f6e\u5206\u8fa8\u7387\u81ea\u52a8\u7f29\u5c0f\u5e76\u5206\u7ec4\u56fe\u50cf\uff0c\u4e0d\u652f\u6301\u653e\u5927\u3002<\/li>\n<\/ul>\n<h4>\u7279\u5b9a\u5c42\u8bad\u7ec3<\/h4>\n<ol>\n<li><strong>\u7f16\u8f91\u914d\u7f6e\u6587\u4ef6<\/strong>: \u5728<code>network<\/code>\u90e8\u5206\u6dfb\u52a0\uff1a<\/li>\n<\/ol>\n<pre><code>network:\r\ntype: \"lora\"\r\nlinear: 128\r\nlinear_alpha: 128\r\nnetwork_kwargs:\r\nonly_if_contains:\r\n- \"<a href=\"https:\/\/www.kdjingpai.com\/pt\/transformer\/\">transformer<\/a>.single_transformer_blocks.7.proj_out\"\r\n- \"transformer.single_transformer_blocks.20.proj_out\"\r\n<\/code><\/pre>\n<ol start=\"2\">\n<li><strong>\u8fd0\u884c\u8bad\u7ec3<\/strong>: \u4f7f\u7528\u4fee\u6539\u540e\u7684\u914d\u7f6e\u6587\u4ef6\u542f\u52a8\uff0c\u4ec5\u8bad\u7ec3\u6307\u5b9a\u5c42\u3002<\/li>\n<\/ol>\n<h3>\u6ce8\u610f\u4e8b\u9879<\/h3>\n<ul>\n<li><strong>\u4e2d\u65ad\u8bad\u7ec3<\/strong>: \u907f\u514d\u5728\u4fdd\u5b58\u68c0\u67e5\u70b9\u65f6\u6309Ctrl+C\uff0c\u4ee5\u514d\u635f\u574f\u6587\u4ef6\u3002<\/li>\n<li><strong>UI\u5b89\u5168\u6027<\/strong>: UI\u76ee\u524d\u4ec5\u5728Linux\u6d4b\u8bd5\uff0c\u5b89\u5168\u6027\u8f83\u4f4e\uff0c\u4e0d\u5efa\u8bae\u66b4\u9732\u4e8e\u516c\u7f51\u3002<\/li>\n<li><strong>\u83b7\u53d6\u5e2e\u52a9<\/strong>: \u53ef\u52a0\u5165Ostris\u7684Discord\u793e\u533a\u63d0\u95ee\uff0c\u907f\u514d\u76f4\u63a5\u79c1\u4fe1\u5f00\u53d1\u8005\u3002<\/li>\n<\/ul>\n","protected":false},"excerpt":{"rendered":"<p>AI Toolkit by Ostris \u662f\u4e00\u4e2a\u5f00\u6e90\u7684AI\u5de5\u5177\u96c6\uff0c\u4e13\u6ce8\u4e8e\u652f\u6301Stable Diffusion\u53caFLUX.1\u6a21\u578b\u7684\u8bad\u7ec3\u4e0e\u56fe\u50cf\u751f\u6210\u4efb\u52a1\u3002\u8be5\u5de5\u5177\u96c6\u7531\u5f00\u53d1\u8005Ostris\u521b\u5efa\u5e76\u7ef4\u62a4\uff0c\u6258\u7ba1\u4e8eGitHub\uff0c\u65e8\u5728\u4e3a\u7814\u7a76\u8005\u548c\u5f00\u53d1\u8005\u63d0\u4f9b\u7075\u6d3b\u7684\u6a21&#8230;<\/p>\n","protected":false},"author":1,"featured_media":62019,"comment_status":"open","ping_status":"closed","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[20],"tags":[222,230,365],"class_list":["post-28303","post","type-post","status-publish","format-standard","has-post-thumbnail","hentry","category-tool","tag-aituxiangshengchengfuzhu","tag-aikaiyuanxiangmu","tag-damoxingweidiao"],"_links":{"self":[{"href":"https:\/\/www.kdjingpai.com\/pt\/wp-json\/wp\/v2\/posts\/28303","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/www.kdjingpai.com\/pt\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/www.kdjingpai.com\/pt\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/www.kdjingpai.com\/pt\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/www.kdjingpai.com\/pt\/wp-json\/wp\/v2\/comments?post=28303"}],"version-history":[{"count":0,"href":"https:\/\/www.kdjingpai.com\/pt\/wp-json\/wp\/v2\/posts\/28303\/revisions"}],"wp:featuredmedia":[{"embeddable":true,"href":"https:\/\/www.kdjingpai.com\/pt\/wp-json\/wp\/v2\/media\/62019"}],"wp:attachment":[{"href":"https:\/\/www.kdjingpai.com\/pt\/wp-json\/wp\/v2\/media?parent=28303"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/www.kdjingpai.com\/pt\/wp-json\/wp\/v2\/categories?post=28303"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/www.kdjingpai.com\/pt\/wp-json\/wp\/v2\/tags?post=28303"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}