Skip to content

Commit 729acea

Browse files
committed
remove is_offloaded.
1 parent 7b73dc2 commit 729acea

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/pipelines/pipeline_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -451,7 +451,7 @@ def module_is_offloaded(module):
451451
f"The module '{module.__class__.__name__}' has been loaded in `bitsandbytes` {'4bit' if is_loaded_in_4bit_bnb else '8bit'} and conversion to {dtype} is not supported. Module is still in {'4bit' if is_loaded_in_4bit_bnb else '8bit'} precision."
452452
)
453453

454-
if is_loaded_in_8bit_bnb and not is_offloaded and device is not None:
454+
if is_loaded_in_8bit_bnb and device is not None:
455455
logger.warning(
456456
f"The module '{module.__class__.__name__}' has been loaded in `bitsandbytes` 8bit and moving it to {device} via `.to()` is not supported. Module is still on {module.device}."
457457
)

0 commit comments

Comments
 (0)