Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat: add FLUX.2 Klein model support

Browse Source 
Add support for FLUX.2 Klein distilled models (4B and 9B variants):

- Add pipeline loader for Flux2KleinPipeline
- Add model detection for 'flux.2' + 'klein' patterns
- Add pipeline mapping in shared_items
- Add shared Qwen3ForCausalLM text encoder handling:
  - 4B variants use Z-Image-Turbo's Qwen3-8B
  - 9B variants use FLUX.2-klein-9B's Qwen3-14B
- Add reference entries for distilled (4B, 9B) and base models
- Update diffusers commit for Flux2KleinPipeline support
This commit is contained in:
CalamitousFelicitousness
2026-01-16 01:16:06 +00:00
parent 6f86cc633f
commit fe99d3fe5d
8 changed files with 106 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_detect.py
+2
View File
@@ -92,6 +92,8 @@ def guess_by_name(fn, current_guess):
new_guess = 'HiDream'
elif 'chroma' in fn.lower() and 'xl' not in fn.lower():
new_guess = 'Chroma'
elif 'flux.2' in fn.lower() and 'klein' in fn.lower():
new_guess = 'FLUX2 Klein'
elif 'flux.2' in fn.lower():
new_guess = 'FLUX2'
elif 'flux' in fn.lower() or 'flex.1' in fn.lower():