DiffusionAi/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2025-01-07 03:17:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update apply_smart_model_patches() so that layer restore matches the behavior of non-smart mode.

Browse Source
This commit is contained in:
Ryan Dick 2024-12-17 17:13:45 +00:00
parent 61253b91f1
commit 6f926f05b0
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
invokeai/backend/patches/model_patcher.py
View File

@ -48,7 +48,8 @@ class LayerPatcher:
finally:
# Restore directly patched layers.
for param_key, weight in original_weights.get_changed_weights():
model.get_parameter(param_key).copy_(weight)
cur_param = model.get_parameter(param_key)
cur_param.data = weight.to(dtype=cur_param.dtype, device=cur_param.device, copy=True)
# Restore LoRASidecarWrapper modules.
# Note: This logic assumes no nested modules in original_modules.