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

Ruff format

Browse Source
This commit is contained in:
Sergey Borisov 2024-07-24 16:01:29 +03:00
parent ab0bfa709a
commit 0ccb304b8b
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/stable_diffusion/extensions/lora_patcher.py
View File

@ -138,7 +138,7 @@ class LoRAPatcherExt(ExtensionBase):
# TODO: debug on lycoris # TODO: debug on lycoris
lora_param_weight = lora_param_weight.reshape(module_param.shape) lora_param_weight = lora_param_weight.reshape(module_param.shape)
lora_param_weight *= (lora_weight * layer_scale) lora_param_weight *= lora_weight * layer_scale
module_param += lora_param_weight.to(dtype=dtype) module_param += lora_param_weight.to(dtype=dtype)
layer.to(device=TorchDevice.CPU_DEVICE) layer.to(device=TorchDevice.CPU_DEVICE)