Skip to content

GitLab

Unverified Commit 6414d4e4 authored by Dhruv Nair's avatar Dhruv Nair Committed by GitHub
Browse files

Fix chunking in SVD (#6350) 

fix
parent 43672b4a
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/models/attention.py
View file @ 6414d4e4
...@@ -498,7 +498,7 @@ class TemporalBasicTransformerBlock(nn.Module): ...@@ -498,7 +498,7 @@ class TemporalBasicTransformerBlock(nn.Module):
hidden_states = self.norm_in(hidden_states) hidden_states = self.norm_in(hidden_states)
if self._chunk_size is not None: 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: else:
hidden_states = self.ff_in(hidden_states) hidden_states = self.ff_in(hidden_states)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment