FLUX.2 Klein 9B LoRA ComfyUI 推理:預覽對齊的 AI Toolkit LoRA 生成在 ComfyUI 中
當您需要在 ComfyUI 中進行推理以保持與 AI Toolkit 訓練預覽一致時,請使用此 RunComfy 工作流程。設置通過 Flux2Klein9BPipeline 生成——由 RunComfy 開源 的模型特定管道封裝——而不是標準取樣器圖形。您的適配器通過該管道中的 lora_path 和 lora_scale 應用,為您提供 訓練匹配 的 LoRA 行為,而無需手動重建管道。
為什麼 FLUX.2 Klein 9B LoRA ComfyUI 推理在 ComfyUI 中看起來不同
當 AI Toolkit 呈現訓練預覽時,它運行完整的 FLUX.2 Klein 9B 推理管道——Qwen3-8B 文本編碼、流匹配調度和內部 LoRA 注入都作為一個協調的單元進行。典型的 ComfyUI 圖形獨立重組這些組件,這會引入條件、噪聲調度和適配器應用順序的微妙差異。結果是管道級漂移,而不是單一錯誤配置的旋鈕。Flux2Klein9BPipeline 通過執行模型自己的管道端到端並在其中注入您的 LoRA 來填補這一空白。參考:`src/pipelines/flux2_klein.py`。
如何使用 FLUX.2 Klein 9B LoRA ComfyUI 推理工作流程
步驟 1:獲取 LoRA 路徑並將其加載到工作流程中(2 個選項)
選項 A — RunComfy 訓練結果 > 下載到本地 ComfyUI:
- 前往 Trainer > LoRA 資產
- 找到您要使用的 FLUX.2 Klein 9B LoRA
- 點擊右側的 ... (三點) 菜單 > 選擇 複製 LoRA 鏈接
- 在 ComfyUI 工作流程頁面 上,將複製的鏈接粘貼到 右上角 UI 的 下載 輸入字段中
- 在點擊下載之前,確保目標文件夾設置為 ComfyUI > models > loras(此文件夾必須被選擇為下載目標)
- 點擊 下載 — LoRA 文件被保存到正確的
models/loras目錄中 - 下載完成後,刷新頁面
- LoRA 現在出現在 LoRA 選擇下拉菜單中 — 選擇它

選項 B — 直接 LoRA URL(覆蓋選項 A):
- 將 直接
.safetensors下載 URL 粘貼到 LoRA 節點的path / url輸入字段中 - 當此處提供 URL 時,它覆蓋選項 A — 工作流程在運行時直接從 URL 獲取 LoRA
- 不需要本地下載或文件放置
提示:確認 URL 解決為實際的 .safetensors 文件,而不是登陸頁面或重定向。

步驟 2:將推理參數與您的訓練樣本設置匹配
在 LoRA 節點上設置 lora_scale 以控制適配器強度——從您在訓練預覽中使用的值開始,然後從那裡調整。
其餘參數位於 生成 和 加載管道 節點上:
prompt— 您的文本提示;包括訓練中的任何觸發詞width/height— 輸出分辨率;匹配您的訓練預覽大小以便直接比較(16 的倍數)sample_steps— 推理步驟;FLUX.2 Klein 9B 默認為 25guidance_scale— CFG 強度;默認為 4.0(Klein 9B 不是指導蒸餾的,因此此值直接影響輸出質量)seed— 固定種子以重現特定輸出;更改它以探索變化seed_mode—fixed或randomizehf_token— 需要有效的 Hugging Face 令牌,因為 FLUX.2 Klein 9B 是一個門控模型;將您的令牌粘貼到加載管道節點的hf_token字段中
訓練對齊提示:如果您在訓練期間自定義了採樣值(種子,guidance_scale,sample_steps,觸發詞),請將這些精確值複製到相應字段中。如果您在 RunComfy 上進行訓練,請打開 Trainer > LoRA 資產 > 配置以查看解析的 YAML 並轉移預覽/樣本設置。

步驟 3:運行 FLUX.2 Klein 9B LoRA ComfyUI 推理
點擊 Queue/Run — SaveImage 節點將結果寫入您的 ComfyUI 輸出文件夾。
⚠️ 重要 · 需要 FLUX.2 訪問和 Hugging Face 令牌
FLUX.2 Klein 9B 模型需要 明確的訪問授權 在 Hugging Face 上。
在運行此工作流程之前:
- 確保您的 Hugging Face 帳戶已 獲得 FLUX.2 (Klein 9B) 的訪問權限
- 創建一個 Hugging Face 訪問令牌
- 將您的令牌粘貼到 RC FLUX.2 節點中的
hf_token字段中
沒有有效的令牌和適當的模型訪問,工作流程 將無法運行。
有關分步說明,請參見: https://www.runcomfy.com/trainer/ai-toolkit/huggingface-token-flux-ostris-ai-toolkit
快速檢查清單:
- ✅ Hugging Face 帳戶具有 FLUX.2 Klein 9B 訪問權限 並且有效的令牌在
hf_token中 - ✅ LoRA 是:下載到
ComfyUI/models/loras(選項 A),或通過直接.safetensorsURL 加載(選項 B) - ✅ 本地下載後刷新頁面(僅限選項 A)
- ✅ 推理參數與訓練
sample配置匹配(如果自定義)
如果以上所有內容都正確,這裡的推理結果應該與您的訓練預覽非常接近。
FLUX.2 Klein 9B LoRA ComfyUI 推理疑難排解
大多數“訓練預覽與 ComfyUI 推理”差距在 FLUX.2 Klein 9B 上來自 管道級別的差異(文本編碼器路徑、調度器/條件設置以及適配器的應用位置和方式)。 RunComfy 工作流程通過在 Flux2Klein9BPipeline 中運行生成並在該管道內注入 LoRA,避免手動重建管道,這是最接近在 ComfyUI 中重現 AI Toolkit 預覽行為的方法。
(1) 401 客戶端錯誤。
為什麼會發生這種情況 FLUX.2 Klein 9B 是一個 門控 Hugging Face 模型。如果您的帳戶沒有訪問權限,或者沒有提供有效的令牌,模型權重無法下載,推理失敗並顯示 401 錯誤。
如何解決
- 確保您的 Hugging Face 帳戶已 獲得對
black-forest-labs/FLUX.2-klein-base-9B的訪問權限。 - 創建一個 Hugging Face 訪問令牌並將其粘貼到 加載管道 節點上的
hf_token字段中。 - 在確認訪問和令牌後,通過 RunComfy AI Toolkit 管道節點 運行推理,以便身份驗證和模型加載在一個一致的管道中發生。
- 有關分步說明,請參見:https://www.runcomfy.com/trainer/ai-toolkit/huggingface-token-flux-ostris-ai-toolkit
(2) Flux 2 Klein 模型 CLIPLoader 錯誤
為什麼會發生這種情況 這些錯誤是由於 文本編碼器不匹配 引起的——例如,加載不兼容的編碼器或混合 Klein 4B 和 Klein 9B 編碼器資產。這通常顯示為 CLIP/文本編碼器加載期間的嵌入或詞彙大小不匹配。
如何解決
- 更新 ComfyUI 到最新版本 以確保 FLUX.2 Klein 支持完整。
- 確保使用 Klein 9B 的正確文本編碼器(Klein 9B 需要 Qwen3-8B;使用 4B 編碼器將失敗)。
- 為了與預覽對齊的 LoRA 推理,請優先使用 RunComfy 管道封裝,它加載正確的編碼器並在 AI Toolkit 預覽使用的相同管道中應用 LoRA。
(3) mat1 和 mat2 的形狀無法相乘 (512x2560 和 7680x3072)
為什麼會發生這種情況 此錯誤表示 條件維度不匹配,通常是由於使用了錯誤的編碼器或對 FLUX.2 Klein 9B 使用了不正確的剪輯/條件類型。模型接收到錯誤形狀的嵌入,導致在採樣期間矩陣乘法失敗。
如何解決
- 如果手動構建圖形,請驗證您使用的是 FLUX.2 Klein 特定文本編碼器,並且 剪輯/條件類型 與 FLUX.2 Klein 的預期相符。
- 為了最可靠的解決方案,通過 RunComfy FLUX.2 Klein 9B 管道封裝 (
model_type = flux2_klein_9b) 運行推理,並通過lora_path注入您的 LoRA。這保持了整個推理堆棧——編碼器、調度器和適配器——與 AI Toolkit 預覽的管道對齊。
現在運行 FLUX.2 Klein 9B LoRA ComfyUI 推理
加載工作流程,粘貼您的 lora_path,輸入有效的 hf_token,讓 Flux2Klein9BPipeline 保持 ComfyUI 輸出與您的 AI Toolkit 訓練預覽對齊。

