Skip to content

Commit c426a34

Browse files
committed
update
1 parent 6a9a3e5 commit c426a34

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/diffusers/hooks/group_offloading.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -164,9 +164,11 @@ def _apply_group_offloading_diffusers_block(
164164

165165
# Always keep parameters and buffers on onload_device
166166
for name, param in module.named_parameters(recurse=False):
167-
param.data = param.data.to(onload_device)
167+
if torch.is_tensor(param.data):
168+
param.data = param.data.to(onload_device)
168169
for name, buffer in module.named_buffers(recurse=False):
169-
buffer.data = buffer.data.to(onload_device)
170+
if torch.is_tensor(buffer.data):
171+
buffer.data = buffer.data.to(onload_device)
170172

171173

172174
def _apply_group_offloading_group_patterns(

0 commit comments

Comments
 (0)