Update lora.py

lora.py

@@ -1,4 +1,3 @@
-from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 import math
 import torch
 import torch.nn as nn
@@ -94,4 +93,4 @@ class LoRAModule(nn.Module):
         lora_output = self.lora_up(lora_output)
 
         # Combine with original output
-        return self.org_forward(x) + lora_output * self.multiplier * self.scale
+        return self.org_forward(x) + lora_output * self.multiplier * self.scale