From c9f9935d8a80b71a2ab5dfd33b5bcc6e39bc32be Mon Sep 17 00:00:00 2001 From: kabachuha Date: Tue, 25 Mar 2025 20:59:27 +0300 Subject: [PATCH] rename original forward to _forward --- comfy/ldm/flux/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/ldm/flux/model.py b/comfy/ldm/flux/model.py index 3638a668a..5026f6f65 100644 --- a/comfy/ldm/flux/model.py +++ b/comfy/ldm/flux/model.py @@ -209,7 +209,7 @@ class Flux(nn.Module): def forward(self, x, timestep, context, y, guidance=None, control=None, transformer_options={}, **kwargs): return comfy.patcher_extension.WrapperExecutor.new_class_executor( - self.forward_lame, + self._forward, self, comfy.patcher_extension.get_all_wrappers(comfy.patcher_extension.WrappersMP.DIFFUSION_MODEL, transformer_options) ).execute(x, timestep, context, y, guidance=guidance, control=control, transformer_options=transformer_options, **kwargs)