From 70ed668f5eb36af76e29d417a2ae1e7d53b00e64 Mon Sep 17 00:00:00 2001 From: Dhruv Nair Date: Wed, 27 Dec 2023 17:37:41 +0530 Subject: [PATCH] Fix chunking in SVD (#6350) fix --- src/diffusers/models/attention.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/models/attention.py b/src/diffusers/models/attention.py index a34d7421b4f94..804c34d617d3a 100644 --- 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)