docs(mm): add comment in move_model_to_device

2024-08-30 20:32:17 +00:00 · 2024-06-03 10:56:04 +10:00 · 2024-06-03 10:56:04 +10:00 · e7513f6088
commit e7513f6088
parent c7f22b6a3b
1 changed files with 1 additions and 0 deletions
--- a/invokeai/backend/model_manager/load/model_cache/model_cache_default.py
+++ b/invokeai/backend/model_manager/load/model_cache/model_cache_default.py
@ -261,6 +261,7 @@ class ModelCache(ModelCacheBase[AnyModel]):
        if torch.device(source_device).type == torch.device(target_device).type:
            return

+        # Some models don't have a `to` method, in which case they run in RAM/CPU.
        if not hasattr(cache_entry.model, "to"):
            return