Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +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
installer.py
+1 -1
View File
@@ -648,7 +648,7 @@ def check_diffusers():
t_start = time.time()
if args.skip_all:
return
sha = '5efb81fa711863fdece9136ad10788440e658b40' # diffusers commit hash
sha = '61f175660a8ac54f1470a74a810e6c38fb4795d5' # diffusers commit hash
# if args.use_rocm or args.use_zluda or args.use_directml:
# sha = '043ab2520f6a19fce78e6e060a68dbc947edb9f9' # lock diffusers versions for now
pkg = pkg_resources.working_set.by_key.get('diffusers', None)