Update modeling_super_linear.py
Browse files- modeling_super_linear.py +1 -1
modeling_super_linear.py
CHANGED
|
@@ -607,7 +607,7 @@ class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
|
|
| 607 |
if x.shape[1] == 1:
|
| 608 |
upsample = upsample.squeeze(-1)
|
| 609 |
|
| 610 |
-
print(f"Upsampled shape: {upsample.shape}")
|
| 611 |
return upsample
|
| 612 |
|
| 613 |
|
|
|
|
| 607 |
if x.shape[1] == 1:
|
| 608 |
upsample = upsample.squeeze(-1)
|
| 609 |
|
| 610 |
+
#print(f"Upsampled shape: {upsample.shape}")
|
| 611 |
return upsample
|
| 612 |
|
| 613 |
|