MCPcopy Create free account
hub / github.com/kyegomez/BitNet / __init__

Method __init__

bitnet/at.py:45–49  ·  view source on GitHub ↗
(self, net, max_seq_len=2048, pad_value=0)

Source from the content-addressed store, hash-verified

43 """
44
45 def __init__(self, net, max_seq_len=2048, pad_value=0):
46 super().__init__()
47 self.max_seq_len = max_seq_len
48 self.pad_value = pad_value
49 self.net = net
50
51 @torch.no_grad()
52 @eval_decorator

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected