wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-16 16:32:34 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Set Flux models to eval mode to restore FP8 performance

Browse Source 
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.
This commit is contained in:
matthew 2025-10-13 14:34:36 -04:00
parent d68ece7301
commit f87480188a
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
comfy/model_base.py
View File

@ -834,6 +834,7 @@ class PixArt(BaseModel):
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):