Feature Extraction
Transformers
Safetensors
Diffusers
English
qwen3
flux
text-encoder
pruning
distillation
Instructions to use SearchingMan/FLUX.2-klein-9B-Text-Encoder-Pruned-5.1B with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use SearchingMan/FLUX.2-klein-9B-Text-Encoder-Pruned-5.1B with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="SearchingMan/FLUX.2-klein-9B-Text-Encoder-Pruned-5.1B")# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("SearchingMan/FLUX.2-klein-9B-Text-Encoder-Pruned-5.1B") model = AutoModel.from_pretrained("SearchingMan/FLUX.2-klein-9B-Text-Encoder-Pruned-5.1B") - Diffusers
How to use SearchingMan/FLUX.2-klein-9B-Text-Encoder-Pruned-5.1B with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("SearchingMan/FLUX.2-klein-9B-Text-Encoder-Pruned-5.1B", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
Initial release: pruned 5.1B text encoder for FLUX.2-klein-9B (bitwise-verified export, benchmarks, seed study)
300ccd6 verified | { | |
| "torch_equal": true, | |
| "max_abs_diff": 0.0, | |
| "gate_prompts": 64, | |
| "gate_prompts_sha256": "00208be2914ca3c39b6e7d85cc6703618b2e8f17ecffdbfd8c1a99b4058599e3", | |
| "gate_batch": 4, | |
| "device": "cuda", | |
| "torch": "2.11.0+cu128", | |
| "transformers": "5.7.0", | |
| "variant": "A_keep_last_layer", | |
| "elapsed_sec": 172.7 | |
| } |