musabc commited on
Commit
eba4f8d
·
verified ·
1 Parent(s): 9d41980

Upload model_v5.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. model_v5.py +8 -2
model_v5.py CHANGED
@@ -212,8 +212,12 @@ class GPTV5(nn.Module):
212
  if targets is not None:
213
  logits = self.lm_head(x)
214
  if self.cfg.logit_softcap > 0:
 
 
215
  cap = self.cfg.logit_softcap
216
- logits = cap * torch.tanh(logits / cap)
 
 
217
  flat_logits = logits.view(-1, logits.size(-1))
218
  flat_targets = targets.view(-1)
219
  if self._ce_loss is not None:
@@ -226,7 +230,9 @@ class GPTV5(nn.Module):
226
  logits = self.lm_head(x[:, [-1], :])
227
  if self.cfg.logit_softcap > 0:
228
  cap = self.cfg.logit_softcap
229
- logits = cap * torch.tanh(logits / cap)
 
 
230
  return logits, None
231
 
232
  @torch.no_grad()
 
212
  if targets is not None:
213
  logits = self.lm_head(x)
214
  if self.cfg.logit_softcap > 0:
215
+ # In-place softcap — bellek tasarrufu (compile yokken kritik)
216
+ # cap * tanh(logits / cap) — 3 tensor yerine in-place
217
  cap = self.cfg.logit_softcap
218
+ logits.div_(cap)
219
+ logits = torch.tanh(logits)
220
+ logits.mul_(cap)
221
  flat_logits = logits.view(-1, logits.size(-1))
222
  flat_targets = targets.view(-1)
223
  if self._ce_loss is not None:
 
230
  logits = self.lm_head(x[:, [-1], :])
231
  if self.cfg.logit_softcap > 0:
232
  cap = self.cfg.logit_softcap
233
+ logits.div_(cap)
234
+ logits = torch.tanh(logits)
235
+ logits.mul_(cap)
236
  return logits, None
237
 
238
  @torch.no_grad()