From c426a343b78d0d2b6aeef54e23457e6046c36790 Mon Sep 17 00:00:00 2001 From: Aryan Date: Fri, 10 Jan 2025 02:51:05 +0100 Subject: [PATCH] update --- src/diffusers/hooks/group_offloading.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/diffusers/hooks/group_offloading.py b/src/diffusers/hooks/group_offloading.py index 48bbdb78e33f..8eda18053eb9 100644 --- a/src/diffusers/hooks/group_offloading.py +++ b/src/diffusers/hooks/group_offloading.py @@ -164,9 +164,11 @@ def _apply_group_offloading_diffusers_block( # Always keep parameters and buffers on onload_device for name, param in module.named_parameters(recurse=False): - param.data = param.data.to(onload_device) + if torch.is_tensor(param.data): + param.data = param.data.to(onload_device) for name, buffer in module.named_buffers(recurse=False): - buffer.data = buffer.data.to(onload_device) + if torch.is_tensor(buffer.data): + buffer.data = buffer.data.to(onload_device) def _apply_group_offloading_group_patterns(