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 |
|