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

Format by black

Browse Source
This commit is contained in:
Sergey Borisov 2023-08-11 03:20:56 +03:00
parent 56023bc725
commit c0c2ab3dcf
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
invokeai/backend/model_management/models/lora.py
View File

@ -393,6 +393,7 @@ class FullLayer(LoRALayerBase):
self.weight = self.weight.to(device=device, dtype=dtype) self.weight = self.weight.to(device=device, dtype=dtype)
class IA3Layer(LoRALayerBase): class IA3Layer(LoRALayerBase):
# weight: torch.Tensor # weight: torch.Tensor
# on_input: torch.Tensor # on_input: torch.Tensor