Update comment.
Browse files- tea_model.py +1 -1
tea_model.py
CHANGED
|
@@ -56,7 +56,7 @@ class TeaDecoder(nn.Module):
|
|
| 56 |
|
| 57 |
|
| 58 |
def forward(self, x):
|
| 59 |
-
# Clamp the input values to a specific range
|
| 60 |
clamped = torch.tanh(x / 1)
|
| 61 |
cooked = self.middle(self.block_in(clamped))
|
| 62 |
|
|
|
|
| 56 |
|
| 57 |
|
| 58 |
def forward(self, x):
|
| 59 |
+
# Clamp the input values to a specific range.
|
| 60 |
clamped = torch.tanh(x / 1)
|
| 61 |
cooked = self.middle(self.block_in(clamped))
|
| 62 |
|