From cac36d93274471606193019b21166abe51e93da4 Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Thu, 18 Jul 2024 19:25:32 -0400 Subject: [PATCH] Update invokeai/backend/model_manager/load/model_cache/model_cache_default.py Co-authored-by: Ryan Dick --- .../model_manager/load/model_cache/model_cache_default.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/invokeai/backend/model_manager/load/model_cache/model_cache_default.py b/invokeai/backend/model_manager/load/model_cache/model_cache_default.py index b1bfe80206..ff9d8e918e 100644 --- a/invokeai/backend/model_manager/load/model_cache/model_cache_default.py +++ b/invokeai/backend/model_manager/load/model_cache/model_cache_default.py @@ -292,7 +292,7 @@ class ModelCache(ModelCacheBase[AnyModel]): try: cache_entry.model.to(target_device, non_blocking=TorchDevice.get_non_blocking(target_device)) except TypeError as e: - if "got an unexpected keyword argument 'non_blocking' in str(e)": + if "got an unexpected keyword argument 'non_blocking'" in str(e): cache_entry.model.to(target_device) else: raise e