fix(mm): fix ModelCacheBase method name

invoke-ai · Feb 26, 2024 · 3ccb4e6 · 3ccb4e6
1 parent 200a9d1
commit 3ccb4e6
Showing 1 changed file with 1 addition and 1 deletion.
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
@@ -111,7 +111,7 @@ def offload_unlocked_models(self, size_required: int) -> None:
         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