From 223a946aeb8f8f5ca55b7ac7e55b964a27ceeca1 Mon Sep 17 00:00:00 2001 From: Marc Sun Date: Mon, 4 Dec 2023 23:58:20 +0100 Subject: [PATCH] fix log --- src/accelerate/big_modeling.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/accelerate/big_modeling.py b/src/accelerate/big_modeling.py index dd0ed1047d0..b2852ec20fc 100644 --- a/src/accelerate/big_modeling.py +++ b/src/accelerate/big_modeling.py @@ -412,7 +412,7 @@ def dispatch_model( # warn if there is any params on the meta device offloaded_devices_str = " and ".join( - [device for device in set(device_map.values()) if device in ("cpu", "disk")] + [device for device in set(device_map.values()) if device in offloaded_devices] ) if len(offloaded_devices_str) > 0: logging.warning(