Fix chunking in SVD (huggingface#6350)

fix
donhardman · Dec 29, 2023 · 70ed668 · 70ed668
1 parent 6893b27
commit 70ed668
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/diffusers/models/attention.py b/src/diffusers/models/attention.py
@@ -498,7 +498,7 @@ def forward(
         hidden_states = self.norm_in(hidden_states)
 
         if self._chunk_size is not None:
-            hidden_states = _chunked_feed_forward(self.ff, hidden_states, self._chunk_dim, self._chunk_size)
+            hidden_states = _chunked_feed_forward(self.ff_in, hidden_states, self._chunk_dim, self._chunk_size)
         else:
             hidden_states = self.ff_in(hidden_states)