Duplicate from Lakonik/AsymFLUX.2-klein-9B

Co-authored-by: Hansheng Chen <Lakonik@users.noreply.huggingface.co>

.gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

LICENSE.md

@@ -0,0 +1,11 @@
+# License for AsymFLUX.2-klein
+
+This repository distributes an **AsymFLUX.2-klein adapter** that is a **Derivative** of
+**FLUX.2 klein Base 9B** by **Black Forest Labs Inc.**
+
+Use and distribution of this app are governed by the **FLUX Non-Commercial License**.
+No commercial use of this app or its derivatives is permitted without a separate
+commercial license from Black Forest Labs.
+
+- Full license: https://huggingface.co/black-forest-labs/FLUX.2-klein-base-9B/blob/main/LICENSE.md
+- This repository does not grant any rights beyond the license above.

README.md

@@ -0,0 +1,93 @@
+---
+base_model:
+- black-forest-labs/FLUX.2-klein-base-9B
+library_name: diffusers
+license: other
+license_name: flux-non-commercial-license
+license_link: LICENSE.md
+pipeline_tag: text-to-image
+tags:
+- flow-matching
+- pixel-diffusion
+- pixel-generation
+- flux2
+---
+
+# Asymmetric Flow Models
+
+Pixel-space text-to-image model AsymFLUX.2-klein finetuned from [black-forest-labs/FLUX.2-klein-base-9B](https://huggingface.co/black-forest-labs/FLUX.2-klein-base-9B), using the AsymFlow method proposed in the paper:
+
+**Asymmetric Flow Models**
+<br>
+arXiv 2026
+<br>
+[Hansheng Chen](https://lakonik.github.io/),
+[Jan Ackermann](https://janackermann.info/),
+[Minseo Kim](https://soniaminseokim.github.io/),
+[Gordon Wetzstein](http://web.stanford.edu/~gordonwz/), 
+[Leonidas Guibas](https://geometry.stanford.edu/?member=guibas)<br>
+Stanford University
+<br>
+[Project Page](https://hanshengchen.com/asymflow) | [arXiv](https://arxiv.org/abs/2605.12964) | [Code](https://github.com/Lakonik/LakonLab/blob/main/docs/AsymFlow.md) | [AsymFLUX.2 klein Demo🤗](https://huggingface.co/spaces/Lakonik/AsymFLUX.2-klein)
+
+![asymflow_teaser](https://cdn-uploads.huggingface.co/production/uploads/638067fcb334960c987fbeda/UCU9seMTK_iBccdFNErns.jpeg)
+
+## Usage
+
+Please first install the [LakonLab v0.2](https://github.com/Lakonik/LakonLab).
+
+We provide a Diffusers-style pipeline for AsymFLUX.2 klein. The example below loads the FLUX.2 klein Base 9B model, attaches the AsymFlow adapter, and generates an image directly in pixel space.
+
+```python
+import math
+import torch
+from lakonlab.models.architectures import OklabColorEncoder
+from lakonlab.models.diffusions.schedulers import FlowAdapterScheduler
+from lakonlab.pipelines.pipeline_pixelflux2_klein import PixelFlux2KleinPipeline
+
+pipe = PixelFlux2KleinPipeline.from_pretrained(
+    'black-forest-labs/FLUX.2-klein-base-9B',
+    vae=OklabColorEncoder(
+        use_affine_norm=True,
+        mean=(0.56, 0.0, 0.01),
+        std=0.16),
+    scheduler=FlowAdapterScheduler(
+        shift=17.0,
+        use_dynamic_shifting=True,
+        base_seq_len=1024 ** 2,
+        max_seq_len=2048 ** 2,
+        base_logshift=math.log(17.0),
+        max_logshift=math.log(34.0),
+        dynamic_shifting_type='sqrt',
+        base_scheduler='UniPCMultistep'),
+    torch_dtype=torch.bfloat16)
+adapter_name = pipe.load_lakonlab_adapter(  # you may later call `pipe.set_adapters([adapter_name, ...])` to combine other adapters (e.g., style LoRAs)
+    'Lakonik/AsymFLUX.2-klein-9B',
+    target_module_name='transformer')
+pipe = pipe.to('cuda')
+
+# Text-to-image generation example
+prompt = 'Restored color photo from the 1900s. A middle-aged man with cybernetic metal hands is sitting on an old wooden chair and reading the newspaper. The newspaper has the prominent headline "AsymFLOW RELEASED" in large bold font. Close-up shot focusing on the newspaper.'
+neg_prompt = 'Low quality, worst quality, blurry, deformed, bad anatomy, unclear text'
+out = pipe(
+    prompt=prompt,
+    negative_prompt=neg_prompt,
+    width=960,
+    height=1280,
+    num_inference_steps=38,
+    guidance_scale=4.0,
+    generator=torch.Generator().manual_seed(42),
+).images[0]
+out.save('asymflux2_klein.png')
+```
+
+## Citation
+```
+@article{chen2026asymmetric,
+  title={Asymmetric Flow Models},
+  author={Hansheng Chen and Jan Ackermann and Minseo Kim and Gordon Wetzstein and Leonidas Guibas},
+  journal={arXiv preprint arXiv:2605.12964},
+  url={https://arxiv.org/abs/2605.12964},
+  year={2026},
+}
+```

config.json

@@ -0,0 +1,25 @@
+{
+  "_class_name": "AsymFlux2Transformer2DModel",
+  "_diffusers_version": "0.37.0",
+  "attention_head_dim": 128,
+  "axes_dims_rope": [
+    32,
+    32,
+    32,
+    32
+  ],
+  "base_rank": 128,
+  "eps": 1e-06,
+  "guidance_embeds": false,
+  "in_channels": 3,
+  "joint_attention_dim": 12288,
+  "mlp_ratio": 3.0,
+  "num_attention_heads": 32,
+  "num_layers": 8,
+  "num_single_layers": 24,
+  "num_timesteps": 1,
+  "patch_size": 16,
+  "rope_theta": 2000,
+  "sigma_min": 0.0001,
+  "timestep_guidance_channels": 256
+}

diffusion_pytorch_model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1d642c87bc4facb7db9578a6f640112aa64f8187ca4ccae925ff6082ca8db153
+size 707083194