Matthieu Beaumont
Initial ai-toolkit Blitz transformer
224c6e4
|
Raw
History Blame Contribute Delete
682 Bytes
metadata
license_name: flux-non-commercial-license
base_model:
  - black-forest-labs/FLUX.2-klein-9B
base_model_relation: finetune
tags:
  - flux
  - flux2
  - klein
  - blitz
  - ai-toolkit

FLUX.2 Klein 9B Blitz

Minimal ai-toolkit-compatible transformer repo for the Blitz fine-tune.

Use with ai-toolkit:

model:
  name_or_path: "wraps/FLUX.2-klein-9B-Blitz"
  arch: "flux2_klein_9b"
  quantize: true
  qtype: "qfloat8"
  quantize_te: true
  qtype_te: "qfloat8"

This repo contains only the Flux2 Klein 9B transformer file expected by ai-toolkit:

flux-2-klein-base-9b.safetensors

The ComfyUI variants are kept in:

wraps/FLUX.2-klein-9B-Blitz-ComfyUI