softcap 15 to 17

2026-03-08 15:50:16 +00:00
parent 438a26e2c3
commit 216eeb8d6e
1 changed files with 1 additions and 1 deletions
@@ -278,7 +278,7 @@ class GPT(nn.Module):
            x = block(x, ve, cos_sin, self.window_sizes[i])
        x = norm(x)

-        softcap = 15
+        softcap = 17
        logits = self.lm_head(x)
        logits = logits.float()
        logits = softcap * torch.tanh(logits / softcap)