MCPcopy Create free account

hub / github.com/EricLBuehler/candle-lora / functions

Functions289 in github.com/EricLBuehler/candle-lora

⨍ Functions 289◇ Types & classes 175
↓ 32 callersFunctionlinear_no_bias
( d1: usize, d2: usize, vb: VarBuilder, merge: bool, lora_config: LoraConfig, )
candle-lora-transformers/src/with_tracing.rs:152
↓ 25 callersMethoddevice
(&self)
candle-lora-transformers/src/t5.rs:804
↓ 19 callersFunctionlayer_norm
(size: usize, eps: f64, vb: VarBuilder)
candle-lora-transformers/src/bert.rs:250
↓ 17 callersMethodbuild
(self)
candle-lora/src/lib.rs:201
↓ 16 callersMethodforward
(&self, xs: &Tensor)
candle-lora-transformers/src/t5.rs:141
↓ 14 callersFunctionlinear
( d1: usize, d2: usize, vb: VarBuilder, merge: bool, lora_config: LoraConfig, )
candle-lora-transformers/src/with_tracing.rs:116
↓ 12 callersMethodencode
(&mut self, input_ids: &Tensor)
candle-lora-transformers/src/t5.rs:896
↓ 12 callersFunctionfrom_mmaped_safetensors
Load tensors into a VarBuilder backed by a VarMap using MmapedSafetensors. Set `silent` to not show a progress bar.
candle-lora-transformers/src/varbuilder_utils.rs:15
↓ 11 callersFunctionis_ident
(ident: &Ident, name: &str)
candle-lora-macro/src/lib.rs:141
↓ 11 callersMethodweight
(&self)
candle-lora/src/loraconv2d.rs:216
↓ 9 callersMethodshape
(&self)
candle-lora-transformers/src/llama.rs:120
↓ 8 callersFunctionconv2d
( c_in: usize, c_out: usize, ksize: usize, padding: usize, stride: usize, vb: VarBuild
candle-lora-transformers/src/resnet.rs:27
↓ 8 callersFunctionlinear
(size1: usize, size2: usize, vb: VarBuilder)
candle-lora-transformers/src/llama.rs:179
↓ 7 callersMethodconfig
(&self)
candle-lora/src/loraconv2d.rs:210
↓ 6 callersFunctionbottleneck_resnet
( nclasses: Option<usize>, c1: usize, c2: usize, c3: usize, c4: usize, vb: VarBuilder,
candle-lora-transformers/src/resnet.rs:341
↓ 6 callersFunctionlinear
( size1: usize, size2: usize, vb: VarBuilder, merge: bool, lora_config: LoraConfig, )
candle-lora-transformers/src/bert.rs:216
↓ 6 callersMethodweight
(&self)
candle-lora/src/loralinear.rs:161
↓ 5 callersMethoddecode
( &mut self, decoder_input_ids: &Tensor, encoder_output: &Tensor, )
candle-lora-transformers/src/t5.rs:900
↓ 5 callersFunctionlinear
(vb: VarBuilder, in_dim: usize, out_dim: usize, bias: bool)
candle-lora-transformers/src/dinov2.rs:16
↓ 5 callersFunctionlinear
( size1: usize, size2: usize, bias: bool, vb: VarBuilder, merge: bool, lora_config: Lo
candle-lora-transformers/src/bigcode.rs:40
↓ 5 callersFunctionlinear
(size1: usize, size2: usize, bias: bool, vb: VarBuilder)
candle-lora-transformers/src/falcon.rs:14
↓ 5 callersMethodshape
(&self)
candle-lora/src/lib.rs:238
↓ 5 callersMethodtranspose_for_scores
(&self, xs: &Tensor)
candle-lora-transformers/src/blip_text.rs:120
↓ 5 callersMethodweight
(&self)
candle-lora/src/loraconv1d.rs:182
↓ 4 callersFunctionbasic_layer
( c_in: usize, c_out: usize, stride: usize, cnt: usize, vb: VarBuilder, merge: bool,
candle-lora-transformers/src/resnet.rs:137
↓ 4 callersFunctionbottleneck_layer
( c_in: usize, c_out: usize, stride: usize, cnt: usize, vb: VarBuilder, merge: bool,
candle-lora-transformers/src/resnet.rs:308
↓ 4 callersMethodconfig
(&self)
candle-lora/src/loraconv1d.rs:176
↓ 4 callersFunctionembedding
(vocab_size: usize, hidden_size: usize, vb: VarBuilder)
candle-lora-transformers/src/bert.rs:211
↓ 4 callersMethodmerge_weights
(&mut self)
candle-lora/src/loraembed.rs:92
↓ 4 callersFunctionresnet
( nclasses: Option<usize>, c1: usize, c2: usize, c3: usize, c4: usize, vb: VarBuilder,
candle-lora-transformers/src/resnet.rs:169
↓ 3 callersMethodadd_linear_layers
( mut self, layers: HashMap<T, &'a dyn LinearLayerLike>, linear_config: LoraLinearConf
candle-lora/src/lib.rs:161
↓ 3 callersMethodbias
(&self)
candle-lora/src/loraconv1d.rs:179
↓ 3 callersFunctionlayer_norm
(size: usize, eps: f64, vb: VarBuilder)
candle-lora-transformers/src/bigcode.rs:118
↓ 3 callersFunctionlayer_norm
(size: usize, eps: f64, vb: VarBuilder)
candle-lora-transformers/src/falcon.rs:24
↓ 3 callersMethodtranspose_for_scores
(&self, xs: &Tensor)
candle-lora-transformers/src/bert.rs:433
↓ 2 callersMethodadd_conv1d_layers
( mut self, layers: HashMap<T, &'a dyn Conv1dLayerLike>, conv1d_config: LoraConv1dConf
candle-lora/src/lib.rs:181
↓ 2 callersMethodadd_conv2d_layers
( mut self, layers: HashMap<T, &'a dyn Conv2dLayerLike>, conv2d_config: LoraConv2dConf
candle-lora/src/lib.rs:191
↓ 2 callersMethodadd_embed_layers
( mut self, layers: HashMap<T, &'a dyn EmbeddingLayerLike>, embed_config: LoraEmbeddin
candle-lora/src/lib.rs:171
↓ 2 callersMethodapply_rotary_emb
(&self, x: &Tensor, index_pos: usize)
candle-lora-transformers/src/llama.rs:246
↓ 2 callersMethodbias
(&self)
candle-lora/src/loraconv2d.rs:213
↓ 2 callersMethodbias
(&self)
candle-lora/src/loralinear.rs:158
↓ 2 callersMethodbias
(&self)
candle-lora/src/frozenconv.rs:36
↓ 2 callersMethodbuild_encoder
(&self)
candle-lora-transformers/examples/t5.rs:124
↓ 2 callersMethodconfig
(&self)
candle-lora-transformers/src/falcon.rs:531
↓ 2 callersMethodconfig
(&self)
candle-lora/src/frozenconv.rs:54
↓ 2 callersFunctionconv2d
( in_channels: usize, out_channels: usize, kernel_size: usize, cfg: candle_nn::Conv2dConfig,
candle-lora-transformers/src/with_tracing.rs:209
↓ 2 callersFunctiondownsample
( c_in: usize, c_out: usize, stride: usize, vb: VarBuilder, merge: bool, lora_config:
candle-lora-transformers/src/resnet.rs:76
↓ 2 callersFunctionembedding
( vocab_size: usize, hidden_size: usize, vb: VarBuilder, merge: bool, lora_config: LoraCon
candle-lora-transformers/src/bigcode.rs:82
↓ 2 callersMethodembeddings
(&self)
candle-lora-transformers/src/with_tracing.rs:58
↓ 2 callersMethodembeddings
(&self)
candle-lora/src/loraembed.rs:155
↓ 2 callersMethodforward
(&self, input: &Tensor)
candle-lora/tests/conv1d.rs:25
↓ 2 callersMethodforward
(&self, input: &Tensor)
candle-lora/tests/conv2d_merged.rs:25
↓ 2 callersMethodforward
(&self, input: &Tensor)
candle-lora/tests/embed.rs:25
↓ 2 callersMethodforward
(&self, input: &Tensor)
candle-lora/tests/linear.rs:25
↓ 2 callersMethodforward
(&self, input: &Tensor)
candle-lora/tests/conv1d_merged.rs:25
↓ 2 callersMethodforward
(&self, input: &Tensor)
candle-lora/tests/conv2d.rs:25
↓ 2 callersMethodforward
(&self, input: &Tensor)
candle-lora/tests/linear_merged.rs:25
↓ 2 callersMethodforward
(&self, input: &Tensor)
candle-lora/tests/embed_merged.rs:25
↓ 2 callersMethodget_delta_weight
(&self)
candle-lora/src/loraembed.rs:84
↓ 2 callersMethodget_delta_weight
(&self)
candle-lora/src/loraconv2d.rs:97
↓ 2 callersMethodget_delta_weight
(&self)
candle-lora/src/loralinear.rs:77
↓ 2 callersMethodget_delta_weight
(&self)
candle-lora/src/loraconv1d.rs:85
↓ 2 callersMethodhead_dim
(&self)
candle-lora-transformers/src/stable_lm.rs:46
↓ 2 callersMethodhead_dim
(&self)
candle-lora-transformers/src/falcon.rs:128
↓ 2 callersMethodhidden_size
(&self)
candle-lora/src/loraembed.rs:158
↓ 2 callersMethodload
( vb: VarBuilder, cfg: &Config, merge: bool, lora_config: LoraConfig, )
candle-lora-transformers/src/bigcode.rs:224
↓ 2 callersFunctionrepeat_kv
This is the equivalent of torch.repeat_interleave(x, dim=1, repeats=n_rep). The hidden states go from (batch, num_key_value_heads, seqlen, head_dim) t
candle-lora-transformers/src/mpt.rs:154
↓ 2 callersMethodrepeat_kv
(&self, x: Tensor)
candle-lora-transformers/src/llama.rs:328
↓ 2 callersMethodrepeat_kv
(&self, xs: Tensor)
candle-lora-transformers/src/stable_lm.rs:256
↓ 2 callersMethodrepeat_kv
(&self, xs: Tensor)
candle-lora-transformers/src/mistral.rs:259
↓ 2 callersMethodrotary_ndims
(&self)
candle-lora-transformers/src/stable_lm.rs:50
↓ 2 callersFunctionrotate_half
(xs: &Tensor)
candle-lora-transformers/src/stable_lm.rs:65
↓ 2 callersFunctionrotate_half
(x: &Tensor)
candle-lora-transformers/src/falcon.rs:137
↓ 2 callersFunctionrotate_half
(xs: &Tensor)
candle-lora-transformers/src/mistral.rs:73
↓ 2 callersMethodweight
(&self)
candle-lora/src/frozenlinear.rs:42
↓ 2 callersMethodweight
(&self)
candle-lora/src/frozenconv.rs:33
↓ 1 callersMethodapply_rotary_emb_qkv
( &self, q: &Tensor, k: &Tensor, seqlen_offset: usize, )
candle-lora-transformers/src/stable_lm.rs:91
↓ 1 callersMethodapply_rotary_emb_qkv
( &self, q: &Tensor, k: &Tensor, seqlen_offset: usize, )
candle-lora-transformers/src/mistral.rs:101
↓ 1 callersMethodattn
( &self, query: &Tensor, key: &Tensor, value: &Tensor, attention_mask:
candle-lora-transformers/src/bigcode.rs:267
↓ 1 callersFunctionbasic_block
( c_in: usize, c_out: usize, stride: usize, vb: VarBuilder, merge: bool, lora_config:
candle-lora-transformers/src/resnet.rs:95
↓ 1 callersMethodbias
(&self)
candle-lora/src/frozenlinear.rs:39
↓ 1 callersFunctionbottleneck_block
Bottleneck versions for ResNet 50, 101, and 152.
candle-lora-transformers/src/resnet.rs:250
↓ 1 callersFunctionbuild_alibi_bias
(cfg: &Config)
candle-lora-transformers/src/mpt.rs:230
↓ 1 callersMethodbuild_conditional_generation
(&self)
candle-lora-transformers/examples/t5.rs:135
↓ 1 callersMethodbuild_model_and_tokenizer
(&self)
candle-lora-transformers/examples/bert.rs:54
↓ 1 callersMethodcos_sin
( &mut self, seq_len: usize, device: &Device, dtype: DType, )
candle-lora-transformers/src/falcon.rs:164
↓ 1 callersFunctionembedding
(cfg: &Config, vb: VarBuilder)
candle-lora-transformers/src/llama.rs:188
↓ 1 callersFunctionembedding
(vocab_size: usize, hidden_size: usize, vb: VarBuilder)
candle-lora-transformers/src/falcon.rs:38
↓ 1 callersMethodembeddings
(&self)
candle-lora/src/frozenembed.rs:39
↓ 1 callersMethodembeddings
(&self)
candle-lora/src/lib.rs:306
↓ 1 callersFunctionflash_attn
( q: &Tensor, k: &Tensor, v: &Tensor, softmax_scale: f32, causal: bool, )
candle-lora-transformers/src/llama.rs:230
↓ 1 callersFunctionflash_attn
( q: &Tensor, k: &Tensor, v: &Tensor, softmax_scale: f32, causal: bool, )
candle-lora-transformers/src/stable_lm.rs:163
↓ 1 callersFunctionflash_attn
( q: &Tensor, k: &Tensor, v: &Tensor, softmax_scale: f32, causal: bool, )
candle-lora-transformers/src/mistral.rs:170
↓ 1 callersMethodforward
(&self, xs: &Tensor)
candle-lora-transformers/src/dinov2.rs:116
↓ 1 callersMethodforward
(&self, input: &Tensor)
candle-lora-examples/examples/linear_macro.rs:14
↓ 1 callersMethodforward
(&self, input: &Tensor)
candle-lora-examples/examples/linear_old.rs:20
↓ 1 callersMethodforward
(&self, input: &Tensor)
candle-lora-macro/tests/option_linear.rs:14
↓ 1 callersMethodforward
(&self, input: &Tensor)
candle-lora-macro/tests/linear.rs:15
↓ 1 callersMethodforward
(&self, input: &Tensor)
candle-lora-macro/examples/embed.rs:17
↓ 1 callersMethodforward
(&self, input: &Tensor)
candle-lora-macro/examples/linear.rs:16
next →1–100 of 289, ranked by callers