MCPcopy Create free account

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

Functions289 in github.com/EricLBuehler/candle-lora

Methodforward
(&self, x: &Tensor)
candle-lora/src/frozenconv.rs:42
Methodfrom_weights
( weights: Tensor, bias: Option<Tensor>, vb: VarBuilder, merge: bool,
candle-lora-transformers/src/with_tracing.rs:78
Functionfunc
(f: F)
candle-lora-transformers/src/unsync_func.rs:19
Methodget_tensors
(&self, _accum: &mut HashMap<String, Tensor>)
candle-lora-transformers/src/llama.rs:108
Methodget_tensors
(&self, accum: &mut HashMap<String, Tensor>)
candle-lora/src/loraembed.rs:142
Methodget_tensors
(&self, _accum: &mut HashMap<String, Tensor>)
candle-lora/src/frozenlinear.rs:33
Methodget_tensors
(&self, _accum: &mut HashMap<String, Tensor>)
candle-lora/src/frozenembed.rs:33
Methodget_tensors
(&self, _accum: &mut HashMap<String, Tensor>)
candle-lora/src/lib.rs:226
Methodget_tensors
(&self, accum: &mut HashMap<String, Tensor>)
candle-lora/src/loraconv2d.rs:197
Methodget_tensors
(&self, accum: &mut HashMap<String, Tensor>)
candle-lora/src/loralinear.rs:145
Methodget_tensors
(&self, accum: &mut HashMap<String, Tensor>)
candle-lora/src/loraconv1d.rs:163
Methodget_tensors
(&self, _accum: &mut HashMap<String, Tensor>)
candle-lora/src/frozenconv.rs:48
Methodhidden_size
(&self)
candle-lora/src/lib.rs:309
Methodimage_captioning_large
()
candle-lora-transformers/src/blip.rs:31
Functionlinear
()
candle-lora-macro/tests/linear.rs:21
Functionlinear
()
candle-lora/tests/linear_merged.rs:7
Methodload
(size: usize, eps: f64, vb: VarBuilder)
candle-lora-transformers/src/llama.rs:201
Methodload
(vb: VarBuilder, config: &Config, merge: bool, lora_config: LoraConfig)
candle-lora-transformers/src/bert.rs:327
Methodload
(device: &Device, cfg: &Config)
candle-lora-transformers/src/falcon.rs:152
Methodload
(h: usize, eps: f64, vb: VarBuilder)
candle-lora-transformers/src/t5.rs:130
Methodload
(args: &Args)
candle-lora-transformers/examples/t5.rs:83
Functionmain
()
candle-lora-transformers/examples/llama.rs:102
Functionmain
()
candle-lora-transformers/examples/blip.rs:59
Functionmain
()
candle-lora-transformers/examples/bert.rs:94
Functionmain
()
candle-lora-transformers/examples/resnet.rs:44
Functionmain
()
candle-lora-transformers/examples/stable_lm.rs:173
Functionmain
()
candle-lora-transformers/examples/dinov2.rs:30
Functionmain
()
candle-lora-transformers/examples/bigcode.rs:125
Functionmain
()
candle-lora-transformers/examples/falcon.rs:155
Functionmain
()
candle-lora-transformers/examples/t5.rs:147
Functionmain
()
candle-lora-transformers/examples/mpt.rs:167
Functionmain
()
candle-lora-transformers/examples/mistral.rs:173
Functionmain
()
candle-lora-examples/examples/linear_macro.rs:19
Functionmain
()
candle-lora-examples/examples/linear_old.rs:35
Functionmain
()
candle-lora-macro/examples/embed.rs:22
Functionmain
()
candle-lora-macro/examples/linear.rs:21
Methodmerge_weights
(&mut self)
candle-lora/src/loraconv2d.rs:139
Methodmerge_weights
(&mut self)
candle-lora/src/loralinear.rs:89
Methodmerge_weights
(&mut self)
candle-lora/src/loraconv1d.rs:99
Methodmusicgen_small
https://huggingface.co/facebook/musicgen-small/blob/495da4ad086b3416a27c6187f9239f9fd96f3962/config.json#L184
candle-lora-transformers/src/t5.rs:97
Methodnew
(use_kv_cache: bool, dtype: DType, config: &Config, device: &Device)
candle-lora-transformers/src/llama.rs:137
Methodnew
(cfg: &VisionConfig, vb: VarBuilder)
candle-lora-transformers/src/blip.rs:73
Methodnew
(act: HiddenAct)
candle-lora-transformers/src/bert.rs:28
Methodnew
(dtype: DType, cfg: &Config, dev: &Device)
candle-lora-transformers/src/stable_lm.rs:71
Methodnew
( d1: usize, d2: usize, vb: VarBuilder, merge: bool, lora_config: Lora
candle-lora-transformers/src/with_tracing.rs:19
Methodnew
( vb: VarBuilder, dim: usize, num_heads: usize, qkv_bias: bool, proj_b
candle-lora-transformers/src/dinov2.rs:47
Methodnew
(f: F)
candle-lora-transformers/src/unsync_func.rs:33
Methodnew
(cfg: &Config, vb: VarBuilder, merge: bool, lora_config: LoraConfig)
candle-lora-transformers/src/blip_text.rs:31
Methodnew
(cfg: &Config, vb: VarBuilder, merge: bool, lora_config: LoraConfig)
candle-lora-transformers/src/mpt.rs:60
Methodnew
(size: usize, eps: f64, vb: VarBuilder)
candle-lora-transformers/src/mistral.rs:53
Methodnew
( model: Model, tokenizer: Tokenizer, seed: u64, temp: Option<f64>, to
candle-lora-transformers/examples/stable_lm.rs:33
Methodnew
( model: GPTBigCode, tokenizer: Tokenizer, seed: u64, temp: Option<f64>,
candle-lora-transformers/examples/bigcode.rs:31
Methodnew
( model: Falcon, tokenizer: Tokenizer, generation_options: GenerationOptions,
candle-lora-transformers/examples/falcon.rs:39
Methodnew
( model: Model, tokenizer: Tokenizer, seed: u64, temp: Option<f64>, to
candle-lora-transformers/examples/mpt.rs:31
Methodnew
( model: Mistral, tokenizer: Tokenizer, seed: u64, temp: Option<f64>,
candle-lora-transformers/examples/mistral.rs:33
Methodnew
(num_embeddings: usize, embedding_dim: usize)
candle-lora/src/loraembed.rs:32
Methodnew
(weight: Tensor, bias: Option<Tensor>)
candle-lora/src/frozenlinear.rs:15
Methodnew
(embeddings: &Tensor, hidden_size: usize)
candle-lora/src/frozenembed.rs:15
Methodnew
Create a new LoRA config. - `rank`: The dimensions of low-rank matrices. - `alpha`: Scaling factor for the LoRA signal. - `dropout`: Dropout probabili
candle-lora/src/lib.rs:115
Methodnew
(in_channels: usize, out_channels: usize)
candle-lora/src/loraconv2d.rs:32
Methodnew
(in_features: usize, out_features: usize)
candle-lora/src/loralinear.rs:32
Methodnew
(kernel_size: usize, in_channels: usize, out_channels: usize)
candle-lora/src/loraconv1d.rs:33
Methodnew
( weight: &Tensor, bias: Option<&Tensor>, config: Conv1dConfig, )
candle-lora/src/frozenconv.rs:15
Methodnew_from_conv1d
(old: &dyn Conv1dLayerLike)
candle-lora/src/frozenconv.rs:29
Methodnew_from_conv2d
(old: &dyn Conv2dLayerLike)
candle-lora/src/frozenconv.rs:86
Methodnew_from_embed
(old: &dyn EmbeddingLayerLike)
candle-lora/src/frozenembed.rs:21
Methodnew_from_linear
(old: &dyn LinearLayerLike)
candle-lora/src/frozenlinear.rs:21
Functionoption_linear
()
candle-lora-macro/tests/option_linear.rs:20
Functionreplace_layer_fields
(_args: TokenStream1, input: TokenStream1)
candle-lora-macro/src/lib.rs:9
Methodreplit_code_v1_5_3b
()
candle-lora-transformers/src/mpt.rs:25
Methodreset_kv_cache
(&mut self)
candle-lora-transformers/src/blip_text.rs:131
Functionresnet101_no_final_layer
( vb: VarBuilder, merge: bool, lora_config: LoraConfig, )
candle-lora-transformers/src/resnet.rs:427
Functionresnet152_no_final_layer
( vb: VarBuilder, merge: bool, lora_config: LoraConfig, )
candle-lora-transformers/src/resnet.rs:444
Functionresnet18_no_final_layer
( vb: VarBuilder, merge: bool, lora_config: LoraConfig, )
candle-lora-transformers/src/resnet.rs:223
Functionresnet34_no_final_layer
( vb: VarBuilder, merge: bool, lora_config: LoraConfig, )
candle-lora-transformers/src/resnet.rs:241
Functionresnet50_no_final_layer
( vb: VarBuilder, merge: bool, lora_config: LoraConfig, )
candle-lora-transformers/src/resnet.rs:410
Methodshape
(&self)
candle-lora/src/frozenlinear.rs:45
Methodshape
(&self)
candle-lora/src/loralinear.rs:164
Functionsingle_linear
()
candle-lora/tests/linear.rs:7
Methodstablelm_3b_4e1t
(use_flash_attn: bool)
candle-lora-transformers/src/stable_lm.rs:28
Methodstarcoder
()
candle-lora-transformers/src/bigcode.rs:196
Methodstarcoder_1b
()
candle-lora-transformers/src/bigcode.rs:151
Methodstarcoder_3b
()
candle-lora-transformers/src/bigcode.rs:166
Methodstarcoder_7b
()
candle-lora-transformers/src/bigcode.rs:181
Methodunmerge_weights
(&mut self)
candle-lora/src/loraembed.rs:109
Methodunmerge_weights
(&mut self)
candle-lora/src/loraconv2d.rs:156
Methodunmerge_weights
(&mut self)
candle-lora/src/loralinear.rs:105
Methodunmerge_weights
(&mut self)
candle-lora/src/loraconv1d.rs:116
Methodweight
(&self)
candle-lora-transformers/src/llama.rs:117
← previous201–289 of 289, ranked by callers