multimodalart HF staff commited on
Commit
40573e7
1 Parent(s): 8079453

Update lora.py

Browse files
Files changed (1) hide show
  1. lora.py +2 -1
lora.py CHANGED
@@ -220,7 +220,8 @@ class LoRAInfModule(LoRAModule):
220
  def default_forward(self, x):
221
  # print("default_forward", self.lora_name, x.size())
222
  org_forward = self.org_forward(x)
223
- lora_up_down = self.lora_up(self.lora_down(x))
 
224
  print(org_forward)
225
  print(lora_up_down)
226
  print(self.multiplier)
 
220
  def default_forward(self, x):
221
  # print("default_forward", self.lora_name, x.size())
222
  org_forward = self.org_forward(x)
223
+ lora_down = self.lora_down(x)
224
+ lora_up_down = self.lora_up(lora_down)
225
  print(org_forward)
226
  print(lora_up_down)
227
  print(self.multiplier)