Update README.md

README.md

@@ -27,11 +27,9 @@ def _get_qwen3_prompt_embeds(...):
   )
   ...
 
-flux2_path = 'black-forest-labs/FLUX.2-klein-base-4B'
-tokenizer = AutoTokenizer.from_pretrained(flux2_path, subfolder='tokenizer')
 vision_model = LightOnVisionModel.from_pretrained('lightonai/LightOnOCR-2-1B')
-embedder = LightingFlux2Klein()
-pipeline = Flux2KleinPipeline.from_pretrained(flux2_path, torch_dtype=torch.bfloat16)
+embedder = LightingFlux2Klein.from_pretrained('nightknocker/flux2-klein-4b-lighting-text-encoder')
+pipeline = Flux2KleinPipeline.from_pretrained('black-forest-labs/FLUX.2-klein-base-4B', torch_dtype=torch.bfloat16)
 ```
 
 ![](images/preview.png)