Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix type error in torch device comparison.

Browse Source
This commit is contained in:
Ryan Dick 2023-09-29 14:23:44 -04:00
parent 123f2b2dbc
commit 7d65555a5a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/backend/model_management/model_cache.py
View File

@ -272,7 +272,7 @@ class ModelCache(object):
source_device = cache_entry.model.device
# Note: We compare device types only so that 'cuda' == 'cuda:0'. This would need to be revised to support
# multi-GPU.
if source_device.type == target_device.type:
if torch.device(source_device).type == torch.device(target_device).type:
return
start_model_to_time = time.time()