We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a8e80dc commit c9fcb7eCopy full SHA for c9fcb7e
python/sglang/multimodal_gen/runtime/layers/lora/linear.py
@@ -59,6 +59,14 @@ def __init__(
59
self.lora_A = None
60
self.lora_B = None
61
62
+ @property
63
+ def weight(self):
64
+ return self.base_layer.weight
65
+
66
67
+ def bias(self):
68
+ return getattr(self.base_layer, "bias", None)
69
70
@torch.compile()
71
def forward(self, x: torch.Tensor) -> torch.Tensor:
72
lora_A = self.lora_A
@@ -79,7 +87,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
79
87
return out + delta, output_bias
80
88
else:
81
89
out, output_bias = self.base_layer(x)
82
- return out.to(x), output_bias
90
+ return out, output_bias
83
91
84
92
def slice_lora_a_weights(self, A: torch.Tensor) -> torch.Tensor:
85
93
return A
0 commit comments