Skip to content

Commit 6414d4e

Browse files
authored
Fix chunking in SVD (#6350)
fix
1 parent 43672b4 commit 6414d4e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/attention.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -498,7 +498,7 @@ def forward(
498498
hidden_states = self.norm_in(hidden_states)
499499

500500
if self._chunk_size is not None:
501-
hidden_states = _chunked_feed_forward(self.ff, hidden_states, self._chunk_dim, self._chunk_size)
501+
hidden_states = _chunked_feed_forward(self.ff_in, hidden_states, self._chunk_dim, self._chunk_size)
502502
else:
503503
hidden_states = self.ff_in(hidden_states)
504504

0 commit comments

Comments
 (0)