Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/kyegomez/BitNet
/ functions
Functions
143 in github.com/kyegomez/BitNet
⨍
Functions
143
◇
Types & classes
33
↳
Endpoints
1
Method
generate
Generates a sequence of tokens based on the input string.
bitnet/inference.py:56
Function
inner
(model, *args, **kwargs)
bitnet/at.py:14
Method
load_model
Loads a trained model from a .pth file.
bitnet/inference.py:41
Method
merge
Merge the LoRa regularization with the weight matrix.
bitnet/bit_lora.py:93
Function
random_tensor
()
tests/tests.py:40
Function
replace_linears_in_pytorch_model
Replaces all instances of nn.Linear in the given model with BitLinear15b. Args: model (nn.Module): The model to modify. Returns
bitnet/replace_hf.py:35
Method
run
(self)
setup.py:205
Method
step
(self, x, cache)
bitnet/bit_mamba.py:182
Method
step
(self, x, cache)
bitnet/bit_mamba.py:403
Function
test_absmax_quantize
()
tests/tests.py:15
Function
test_absmax_quantize_bits
(random_tensor, bits)
tests/tests.py:48
Function
test_bitfeedforward_forward_pass
()
tests/test_bitffn.py:19
Function
test_bitfeedforward_initialization
()
tests/test_bitffn.py:8
Function
test_bitlinear_forward
()
tests/tests.py:29
Function
test_bitlinear_forward_pass
()
tests/test_bitlinear.py:16
Function
test_bitlinear_forward_with_groups
(random_tensor, groups)
tests/tests.py:79
Function
test_bitlinear_groups
(groups)
tests/tests.py:66
Function
test_bitlinear_initialization
()
tests/tests.py:22
Function
test_bitlinear_initialization
()
tests/test_bitlinear.py:6
Function
test_bitlinear_input_quantization
(random_tensor)
tests/tests.py:119
Function
test_bitlinear_no_bias
()
tests/test_bitlinear.py:23
Function
test_bitlinear_quantization
()
tests/test_bitlinear.py:28
Function
test_bitlinear_reset_parameters
()
tests/tests.py:71
Function
test_bitlinear_shapes
(in_features, out_features)
tests/tests.py:60
Function
test_bitlinear_weight_group_normalization
(groups)
tests/tests.py:102
Function
test_bitlinear_weight_group_scaling
()
tests/tests.py:109
Function
test_bitlinear_weight_sign
()
tests/tests.py:92
Function
test_bitlinear_zero_input
()
tests/tests.py:85
Function
test_bitnet_transformer_embed
(bitnet_model)
tests/tests.py:198
Function
test_bitnet_transformer_for_various_input_shapes
(bitnet_model, batch_size, seq_len)
tests/tests.py:228
Function
test_bitnet_transformer_forward
(bitnet_model)
tests/tests.py:184
Function
test_bitnet_transformer_forward_pass
()
tests/test_transformer.py:14
Function
test_bitnet_transformer_initialization
()
tests/test_transformer.py:6
Function
test_bitnet_transformer_logit_values
(bitnet_model)
tests/tests.py:269
Function
test_bitnet_transformer_output_shapes
( bitnet_model, input_value, expected_output_shape )
tests/tests.py:256
Function
test_exceptions_on_wrong_dtype
()
tests/tests.py:263
Function
test_mask_persistency
(random_tensor, mask_value)
tests/tests.py:243
Function
test_mocking_get_mask
(monkeypatch, random_tensor)
tests/tests.py:279
Function
test_parallel_transformer_block
(dim, dim_head, heads, ff_mult, random_tensor)
tests/tests.py:164
Function
test_parallel_transformer_block_masking
(random_tensor)
tests/tests.py:190
Function
test_parallel_transformer_block_raises_for_incorrect_input
( dim, dim_head, heads, ff_mult )
tests/tests.py:212
Function
test_rotary_embedding
(bitnet_model, random_tensor)
tests/tests.py:234
Function
test_transformer
(dim, depth, heads, dim_head, ff_mult, random_tensor)
tests/tests.py:178
← previous
101–143 of 143, ranked by callers