home sampling
View code on Github
මෙන්නමෙම නියැදි ශිල්පීය ක්රම භාවිතා කරන අත්හදා බැලීමක් .
18import torch
21class Sampler:
logits
[..., n_tokens]
25 def __call__(self, logits: torch.Tensor) -> torch.Tensor:
31 raise NotImplementedError()