Update README.md

README.md

@@ -26,6 +26,8 @@ from diffusers import Flux2KleinPipeline, Flux2Transformer2DModel
 from transformers.modeling_utils import no_init_weights
 from dfloat11 import DFloat11Model
 
+text_encoder = DFloat11Model.from_pretrained("DFloat11/Qwen3-4B-DF11", device="cpu")
+
 with no_init_weights():
 	transformer = Flux2Transformer2DModel.from_config(
 		Flux2Transformer2DModel.load_config(
@@ -35,7 +37,7 @@ with no_init_weights():
 	).to(torch.bfloat16)
 DFloat11Model.from_pretrained("mingyi456/FLUX.2-klein-4B-DF11", device="cpu", bfloat16_model=transformer)
 
-pipe = Flux2KleinPipeline.from_pretrained("black-forest-labs/FLUX.2-klein-4B", transformer=transformer, torch_dtype=torch.bfloat16)
+pipe = Flux2KleinPipeline.from_pretrained("black-forest-labs/FLUX.2-klein-4B", text_encoder=text_encoder, transformer=transformer, torch_dtype=torch.bfloat16)
 pipe.to("cuda")
 
 prompt = "A cat holding a sign that says hello world"