MCPcopy Create free account

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

Functions289 in github.com/EricLBuehler/candle-lora

⨍ Functions 289◇ Types & classes 175
↓ 1 callersMethodforward
(&self, input: &Tensor)
candle-lora/src/loraconv2d.rs:175
↓ 1 callersFunctionfrom_npz_tensors
Load tensors into a VarBuilder backed by a VarMap using NpzTensors. Set `silent` to not show a progress bar.
candle-lora-transformers/src/varbuilder_utils.rs:51
↓ 1 callersFunctionfrom_pth_tensors
Load tensors into a VarBuilder backed by a VarMap using PthTensors. Set `silent` to not show a progress bar.
candle-lora-transformers/src/varbuilder_utils.rs:88
↓ 1 callersFunctionget_mask
(size: usize, device: &Device)
candle-lora-transformers/src/t5.rs:27
↓ 1 callersFunctionget_mask
(size: usize, device: &Device)
candle-lora-transformers/src/mpt.rs:319
↓ 1 callersMethodhidden_size
(&self)
candle-lora/src/frozenembed.rs:42
↓ 1 callersMethodinsert_new
(&mut self, new: NewLayers<ModelLayers>)
candle-lora-examples/examples/linear_old.rs:26
↓ 1 callersMethodinsert_new
(&mut self, new: NewLayers<ModelLayers>)
candle-lora/tests/conv1d.rs:31
↓ 1 callersMethodinsert_new
(&mut self, new: NewLayers<ModelLayers>)
candle-lora/tests/conv2d_merged.rs:31
↓ 1 callersMethodinsert_new
(&mut self, new: NewLayers<ModelLayers>)
candle-lora/tests/embed.rs:31
↓ 1 callersMethodinsert_new
(&mut self, new: NewLayers<ModelLayers>)
candle-lora/tests/linear.rs:31
↓ 1 callersMethodinsert_new
(&mut self, new: NewLayers<ModelLayers>)
candle-lora/tests/conv1d_merged.rs:31
↓ 1 callersMethodinsert_new
(&mut self, new: NewLayers<ModelLayers>)
candle-lora/tests/conv2d.rs:31
↓ 1 callersMethodinsert_new
(&mut self, new: NewLayers<ModelLayers>)
candle-lora/tests/linear_merged.rs:31
↓ 1 callersMethodinsert_new
(&mut self, new: NewLayers<ModelLayers>)
candle-lora/tests/embed_merged.rs:31
↓ 1 callersMethodinterpolate_pos_encoding
(&self, xs: &Tensor, w: usize, h: usize)
candle-lora-transformers/src/dinov2.rs:460
↓ 1 callersMethodinto_config
(self, use_flash_attn: bool)
candle-lora-transformers/src/llama.rs:34
↓ 1 callersMethodis_causal
(&self)
candle-lora-transformers/src/mpt.rs:40
↓ 1 callersFunctionload_image
Loads an image from disk using the image crate, this returns a tensor with shape (3, 384, 384). OpenAI normalization is applied.
candle-lora-transformers/examples/blip.rs:42
↓ 1 callersFunctionmake_causal_mask
(t: usize, device: &Device)
candle-lora-transformers/src/bigcode.rs:124
↓ 1 callersFunctionmake_causal_mask
(t: usize)
candle-lora-transformers/src/falcon.rs:515
↓ 1 callersMethodmask
(&self, t: usize)
candle-lora-transformers/src/llama.rs:164
↓ 1 callersFunctionmasked_fill
(on_false: &Tensor, mask: &Tensor, on_true: f32)
candle-lora-transformers/src/llama.rs:398
↓ 1 callersFunctionmasked_fill
(on_false: &Tensor, mask: &Tensor, on_true: f32)
candle-lora-transformers/src/falcon.rs:202
↓ 1 callersFunctionmasked_fill
(on_false: &Tensor, mask: &Tensor, on_true: f32)
candle-lora-transformers/src/t5.rs:34
↓ 1 callersFunctionmasked_fill
(on_false: &Tensor, mask: &Tensor, on_true: f32)
candle-lora-transformers/src/mpt.rs:326
↓ 1 callersFunctionnormalize_l2
(v: &Tensor)
candle-lora-transformers/examples/bert.rs:200
↓ 1 callersFunctionnormalize_l2
(v: &Tensor)
candle-lora-transformers/examples/t5.rs:305
↓ 1 callersMethodnum_kv_groups
(&self)
candle-lora-transformers/src/stable_lm.rs:54
↓ 1 callersFunctionprepare_attn_mask
(b_sz: usize, seq_len: usize)
candle-lora-transformers/src/falcon.rs:523
↓ 1 callersMethodprepare_decoder_attention_mask
( &self, b_size: usize, tgt_len: usize, seqlen_offset: usize, )
candle-lora-transformers/src/stable_lm.rs:448
↓ 1 callersMethodprepare_decoder_attention_mask
( &self, b_size: usize, tgt_len: usize, seqlen_offset: usize, )
candle-lora-transformers/src/mistral.rs:442
↓ 1 callersMethodprepare_tokens_with_mask
(&self, xs: &Tensor)
candle-lora-transformers/src/dinov2.rs:486
↓ 1 callersFunctionresnet101
( num_classes: usize, vb: VarBuilder, merge: bool, lora_config: LoraConfig, )
candle-lora-transformers/src/resnet.rs:418
↓ 1 callersFunctionresnet152
( num_classes: usize, vb: VarBuilder, merge: bool, lora_config: LoraConfig, )
candle-lora-transformers/src/resnet.rs:435
↓ 1 callersFunctionresnet18
Creates a ResNet-18 model.
candle-lora-transformers/src/resnet.rs:214
↓ 1 callersFunctionresnet34
Creates a ResNet-34 model.
candle-lora-transformers/src/resnet.rs:232
↓ 1 callersFunctionresnet50
( num_classes: usize, vb: VarBuilder, merge: bool, lora_config: LoraConfig, )
candle-lora-transformers/src/resnet.rs:401
↓ 1 callersMethodrotary
(&self)
candle-lora-transformers/src/falcon.rs:132
↓ 1 callersMethodrun
(&mut self, prompt: &str, sample_len: usize)
candle-lora-transformers/examples/stable_lm.rs:54
↓ 1 callersMethodrun
(&mut self, prompt: &str, sample_len: usize)
candle-lora-transformers/examples/bigcode.rs:48
↓ 1 callersMethodrun
(&mut self, prompt: &str, sample_len: usize)
candle-lora-transformers/examples/falcon.rs:60
↓ 1 callersMethodrun
(&mut self, prompt: &str, sample_len: usize)
candle-lora-transformers/examples/mpt.rs:54
↓ 1 callersMethodrun
(&mut self, prompt: &str, sample_len: usize)
candle-lora-transformers/examples/mistral.rs:54
↓ 1 callersMethodsplit_heads
(&self, fused_qkv: &Tensor)
candle-lora-transformers/src/falcon.rs:334
↓ 1 callersMethodtext_decoder
(&mut self)
candle-lora-transformers/src/blip.rs:360
↓ 1 callersMethodvalidate
(&self)
candle-lora-transformers/src/falcon.rs:90
↓ 1 callersMethodvision_model
(&self)
candle-lora-transformers/src/blip.rs:356
↓ 1 callersFunctionvit_small
( vb: VarBuilder, merge: bool, lora_config: LoraConfig, )
candle-lora-transformers/src/dinov2.rs:508
↓ 1 callersMethodweight
(&self)
candle-lora/src/lib.rs:232
Method_all_mini_lm_l6_v2
()
candle-lora-transformers/src/bert.rs:188
Functionauto_lora_convert
(tokens: TokenStream1)
candle-lora-macro/src/lib.rs:146
Methodbias
(&self)
candle-lora-transformers/src/llama.rs:114
Methodbias
(&self)
candle-lora/src/lib.rs:235
Methodclear_kv_cache
(&mut self)
candle-lora-transformers/src/t5.rs:511
Methodconfig
(&self)
candle-lora-transformers/src/bigcode.rs:444
Methodconfig
(&self)
candle-lora/src/lib.rs:257
Methodconfig_7b_v0_1
(use_flash_attn: bool)
candle-lora-transformers/src/mistral.rs:28
Methodconfig_7b_v1
(use_flash_attn: bool)
candle-lora-transformers/src/llama.rs:62
Methodconfig_7b_v2
(use_flash_attn: bool)
candle-lora-transformers/src/llama.rs:76
Functionconv1d
()
candle-lora/tests/conv1d.rs:7
Functionconv1d
()
candle-lora/tests/conv1d_merged.rs:7
Functionconv2d
()
candle-lora/tests/conv2d_merged.rs:7
Functionconv2d
()
candle-lora/tests/conv2d.rs:7
Methodconvert_model
Convert the selected layers into their LoRA counterparts.
candle-lora/src/lib.rs:25
Methoddefault
()
candle-lora-transformers/src/bert.rs:165
Methoddefault
()
candle-lora-transformers/src/falcon.rs:66
Methoddefault
()
candle-lora-transformers/src/t5.rs:70
Methoddefault
()
candle-lora/src/lib.rs:140
Functiondefault_is_decoder
()
candle-lora-transformers/src/t5.rs:15
Functiondefault_relative_attention_max_distance
()
candle-lora-transformers/src/t5.rs:11
Functiondefault_rope
()
candle-lora-transformers/src/llama.rs:29
Functiondefault_tie_word_embeddings
()
candle-lora-transformers/src/t5.rs:23
Functiondefault_use_cache
()
candle-lora-transformers/src/t5.rs:19
Methodderef
(&self)
candle-lora-transformers/src/bert.rs:273
Methodderef
(&self)
candle-lora-transformers/src/dinov2.rs:33
Methodderef
(&self)
candle-lora-transformers/src/bigcode.rs:21
Methodderef
(&self)
candle-lora-transformers/src/falcon.rs:224
Functionembed
()
candle-lora/tests/embed.rs:7
Functionembed
()
candle-lora/tests/embed_merged.rs:7
Methodfalcon7b
https://huggingface.co/tiiuae/falcon-7b/blob/main/config.json
candle-lora-transformers/src/falcon.rs:104
Methodfmt
(&self, f: &mut std::fmt::Formatter<'_>)
candle-lora-transformers/src/with_tracing.rs:249
Methodfmt
(&self, f: &mut std::fmt::Formatter)
candle-lora-transformers/src/unsync_func.rs:14
Methodforward
(&self, x: &Tensor)
candle-lora-transformers/src/llama.rs:101
Methodforward
(&self, xs: &Tensor)
candle-lora-transformers/src/blip.rs:100
Methodforward
(&self, xs: &Tensor)
candle-lora-transformers/src/bert.rs:33
Methodforward
(&self, xs: &candle_core::Tensor)
candle-lora-transformers/src/resnet.rs:21
Methodforward
(&self, xs: &Tensor)
candle-lora-transformers/src/stable_lm.rs:154
Methodforward
(&self, xs: &Tensor)
candle-lora-transformers/src/with_tracing.rs:64
Methodforward
(&mut self, hidden_states: &Tensor, attention_mask: &Tensor)
candle-lora-transformers/src/bigcode.rs:316
Methodforward
( &mut self, query: &Tensor, key: &Tensor, past_kv_len: usize, )
candle-lora-transformers/src/falcon.rs:186
Methodforward
(&self, xs: &Tensor)
candle-lora-transformers/src/unsync_func.rs:27
Methodforward
(&self, xs: &Tensor, past_kv_len: usize)
candle-lora-transformers/src/blip_text.rs:57
Methodforward
(&mut self, xs: &Tensor, mask: Option<&Tensor>)
candle-lora-transformers/src/mpt.rs:93
Methodforward
(&self, x: &Tensor)
candle-lora-transformers/src/mistral.rs:61
Methodforward
(&self, input: &Tensor)
candle-lora/src/loraembed.rs:128
Methodforward
(&self, x: &Tensor)
candle-lora/src/frozenlinear.rs:27
Methodforward
(&self, indexes: &Tensor)
candle-lora/src/frozenembed.rs:27
Methodforward
(&self, input: &Tensor)
candle-lora/src/loralinear.rs:123
Methodforward
(&self, input: &Tensor)
candle-lora/src/loraconv1d.rs:135
← previousnext →101–200 of 289, ranked by callers