From f87480188af114f1b65ebb87af888d83690578d4 Mon Sep 17 00:00:00 2001 From: matthew Date: Mon, 13 Oct 2025 14:34:36 -0400 Subject: [PATCH] Set Flux models to eval mode to restore FP8 performance Fixes performance regression introduced in #9854 where the fp8_linear optimization path was being skipped during inference because Flux models were not explicitly set to eval mode. --- comfy/model_base.py | 1 + 1 file changed, 1 insertion(+) diff --git a/comfy/model_base.py b/comfy/model_base.py index b0b9cde7d087..cadeddf6bf6d 100644 --- a/comfy/model_base.py +++ b/comfy/model_base.py @@ -834,6 +834,7 @@ def extra_conds(self, **kwargs): class Flux(BaseModel): def __init__(self, model_config, model_type=ModelType.FLUX, device=None, unet_model=comfy.ldm.flux.model.Flux): super().__init__(model_config, model_type, device=device, unet_model=unet_model) + self.diffusion_model.eval().requires_grad_(False) self.memory_usage_factor_conds = ("ref_latents",) def concat_cond(self, **kwargs):