remove unneeded flux2 text encoder float cast before quantization

kevchengcodes · kevchengcodes · commit a25e79ca6d27 · 2026-07-31T13:13:21.000-07:00
diff --git a/python/src/coreai_models/diffusion/pipeline.py b/python/src/coreai_models/diffusion/pipeline.py
@@ -162,8 +162,6 @@ def _load_hf_pipeline(model_id: str, pipeline_type: str, model_dtype: torch.dtyp
         from diffusers import Flux2KleinPipeline
 
         hf_pipe = Flux2KleinPipeline.from_pretrained(model_id, torch_dtype=model_dtype)
-        # Text encoder needs float32 for token embedding precision
-        hf_pipe.text_encoder = hf_pipe.text_encoder.float()
         return hf_pipe
 
     if pipeline_type == "sd3":
