smzerbe commited on
Commit
e59a1f4
·
verified ·
1 Parent(s): 490d2ca

Update extensions-builtin/Lora/networks.py

Browse files
extensions-builtin/Lora/networks.py CHANGED
@@ -581,7 +581,7 @@ def network_Linear_forward(self, input):
581
 
582
  network_apply_weights(self)
583
 
584
- return originals.Linear_forward(self, input.float())
585
 
586
 
587
  def network_Linear_load_state_dict(self, *args, **kwargs):
 
581
 
582
  network_apply_weights(self)
583
 
584
+ return originals.Linear_forward(self.float(), input.float())
585
 
586
 
587
  def network_Linear_load_state_dict(self, *args, **kwargs):