We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6a9a3e5 commit c426a34Copy full SHA for c426a34
src/diffusers/hooks/group_offloading.py
@@ -164,9 +164,11 @@ def _apply_group_offloading_diffusers_block(
164
165
# Always keep parameters and buffers on onload_device
166
for name, param in module.named_parameters(recurse=False):
167
- param.data = param.data.to(onload_device)
+ if torch.is_tensor(param.data):
168
+ param.data = param.data.to(onload_device)
169
for name, buffer in module.named_buffers(recurse=False):
- buffer.data = buffer.data.to(onload_device)
170
+ if torch.is_tensor(buffer.data):
171
+ buffer.data = buffer.data.to(onload_device)
172
173
174
def _apply_group_offloading_group_patterns(
0 commit comments