Commit
•
acffa62
1
Parent(s):
ae40cb9
fix: remove prints
Browse filesSigned-off-by: jupyterjazz <saba.sturua@jina.ai>
- modeling_lora.py +0 -2
modeling_lora.py
CHANGED
|
@@ -188,7 +188,6 @@ class LoRAParametrization(nn.Module):
|
|
| 188 |
|
| 189 |
out = F.linear(input, weights, self.bias)
|
| 190 |
|
| 191 |
-
print('lin', task_idx, input.shape, out.shape)
|
| 192 |
if residual:
|
| 193 |
return out, input
|
| 194 |
return out
|
|
@@ -219,7 +218,6 @@ class LoRAParametrization(nn.Module):
|
|
| 219 |
input, weights, self.padding_idx, self.max_norm,
|
| 220 |
self.norm_type, self.scale_grad_by_freq, self.sparse)
|
| 221 |
|
| 222 |
-
print('emb', task_idx, input.shape, out.shape)
|
| 223 |
return out
|
| 224 |
|
| 225 |
layer.forward = new_forward.__get__(layer, layer.__class__)
|
|
|
|
| 188 |
|
| 189 |
out = F.linear(input, weights, self.bias)
|
| 190 |
|
|
|
|
| 191 |
if residual:
|
| 192 |
return out, input
|
| 193 |
return out
|
|
|
|
| 218 |
input, weights, self.padding_idx, self.max_norm,
|
| 219 |
self.norm_type, self.scale_grad_by_freq, self.sparse)
|
| 220 |
|
|
|
|
| 221 |
return out
|
| 222 |
|
| 223 |
layer.forward = new_forward.__get__(layer, layer.__class__)
|