From 40476bb8b347eec01cf1e21cae64a6bd258b5ebb Mon Sep 17 00:00:00 2001 From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com> Date: Fri, 2 Aug 2024 04:20:18 -0700 Subject: [PATCH] Update unet.py --- backend/nn/unet.py | 1 - 1 file changed, 1 deletion(-) diff --git a/backend/nn/unet.py b/backend/nn/unet.py index a8c721d6..33e0bf36 100644 --- a/backend/nn/unet.py +++ b/backend/nn/unet.py @@ -202,7 +202,6 @@ class BasicTransformerBlock(nn.Module): self.checkpoint = checkpoint self.n_heads = n_heads self.d_head = d_head - self.switch_temporal_ca_to_sa = switch_temporal_ca_to_sa def forward(self, x, context=None, transformer_options={}): return checkpoint(self._forward, (x, context, transformer_options), self.parameters(), self.checkpoint)