diff --git a/invokeai/backend/model_manager/load/model_cache/model_cache_base.py b/invokeai/backend/model_manager/load/model_cache/model_cache_base.py index 4a4a3c7d29..eb82f87cb2 100644 --- a/invokeai/backend/model_manager/load/model_cache/model_cache_base.py +++ b/invokeai/backend/model_manager/load/model_cache/model_cache_base.py @@ -111,7 +111,7 @@ class ModelCacheBase(ABC, Generic[T]): pass @abstractmethod - def move_model_to_device(self, cache_entry: CacheRecord[AnyModel], device: torch.device) -> None: + def move_model_to_device(self, cache_entry: CacheRecord[AnyModel], target_device: torch.device) -> None: """Move model into the indicated device.""" pass