From 532a16f3b9557f49ef8e3d5dc7978d2040a0f4a1 Mon Sep 17 00:00:00 2001 From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com> Date: Sat, 1 Aug 2026 20:15:51 -0700 Subject: [PATCH] Disable gradients on diffusion models. (#15218) --- comfy/model_base.py | 1 + 1 file changed, 1 insertion(+) diff --git a/comfy/model_base.py b/comfy/model_base.py index ee6dc57a2..7c68c6eba 100644 --- a/comfy/model_base.py +++ b/comfy/model_base.py @@ -169,6 +169,7 @@ class BaseModel(torch.nn.Module): else: operations = model_config.custom_operations self.diffusion_model = unet_model(**unet_config, device=device, operations=operations) + self.diffusion_model.requires_grad_(False) self.diffusion_model.eval() if comfy.model_management.force_channels_last(): self.diffusion_model.to(memory_format=torch.channels_last)