1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

fix: remove duplicated code in TemporalBasicTransformerBlock. (#7212)

fix: remove duplicate code in TemporalBasicTransformerBlock.

Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
This commit is contained in:
Rinne
2024-03-07 15:55:22 +08:00
committed by GitHub
parent b5814c5555
commit ada3bb941b

View File

@@ -440,7 +440,6 @@ class TemporalBasicTransformerBlock(nn.Module):
# Define 3 blocks. Each block has its own normalization layer.
# 1. Self-Attn
self.norm_in = nn.LayerNorm(dim)
self.ff_in = FeedForward(
dim,
dim_out=time_mix_inner_dim,