MCPcopy Create free account

hub / github.com/evilsocket/cake / functions

Functions2,320 in github.com/evilsocket/cake

⨍ Functions 2,320◇ Types & classes 325
↓ 5 callersFunctionmake_vb_moe
(cfg: &Config)
cake-core/tests/unit_tests/test_moe.rs:23
↓ 5 callersFunctionmake_vb_standard_block
Build VarBuilder for a standard block with optional extras.
cake-core/benches/bench_helpers.rs:267
↓ 5 callersMethodmatmul
Matrix multiplication. GPU backends override to use accelerated matmul. Default: delegates to candle's CPU/CUDA/Metal matmul.
cake-core/src/backends/mod.rs:185
↓ 5 callersMethodoutput_modality
(&self)
cake-core/src/cake/sharding/api/test_helpers.rs:137
↓ 5 callersFunctionread_model_name_from_config
Try to read a model name from config.json (e.g. _name_or_path field).
cake-core/src/utils/models.rs:326
↓ 5 callersMethodreset_counter
(&mut self)
cake-core/src/models/vibevoice/vae_decoder.rs:31
↓ 5 callersFunctionrope
(pos: &Tensor, dim: usize, theta: usize)
cake-core/src/models/flux/flux1_model.rs:307
↓ 5 callersMethodscale_shift
(&self, xs: &Tensor)
cake-core/src/models/flux/flux1_model.rs:465
↓ 5 callersFunctionscaled_dot_product_attention
(q: &Tensor, k: &Tensor, v: &Tensor, backend: &dyn ComputeBackend)
cake-core/src/models/flux/flux1_model.rs:262
↓ 5 callersFunctionsine_samples
(n: usize)
cake-core/benches/bench_wav.rs:3
↓ 5 callersFunctiontest_config
()
cake-core/src/models/qwen3_5_moe/moe.rs:391
↓ 4 callersMethodadd_message
(&mut self, message: Message)
cake-core/src/cake/sharding/api/test_helpers.rs:96
↓ 4 callersMethodassign_layers
( &self, workers: &[&dyn super::WorkerCapacity], num_layers: usize, master_tfl
cake-core/src/cake/sharding/default.rs:11
↓ 4 callersFunctionauthenticate_as_worker
Worker side of the mutual authentication. Called on the accepted TCP stream before reading any Cake protocol messages.
cake-core/src/cake/sharding/auth.rs:88
↓ 4 callersMethodcausal_mask
Create a causal attention mask. Returns a U8 tensor of shape `(seq_len, kv_len)` where 1 = masked (future position), 0 = attend. Callers use `masked_f
cake-core/src/backends/mod.rs:533
↓ 4 callersMethodcontains_tensor
(&self, name: &str)
cake-core/src/models/vibevoice/vibevoice_1_5b.rs:47
↓ 4 callersMethoddecode
( forwarder: &mut Box<dyn Forwarder>, latents: Tensor, ctx: &mut Context, )
cake-core/src/models/sd/vae.rs:124
↓ 4 callersFunctiondequantize_gptq_4bit
Dequantize a GPTQ 4-bit quantized weight matrix to F32. Inputs (all on CPU): - `qweight`: int32, shape `(packed_rows, out_features)` where `packed_ro
cake-core/src/utils/gptq.rs:96
↓ 4 callersFunctiondispatch_unary
( s: &candle_core::MetalStorage, l: &Layout, f32_kernel: &'static str, f16_kernel: &'static str, label
cake-core/src/backends/metal/mod.rs:171
↓ 4 callersFunctiondummy_context
()
cake-core/src/cake/sharding/api/test_helpers.rs:266
↓ 4 callersMethodencode
Run the full encoder forward pass, returning hidden states. FLUX.2-klein extracts hidden states from layers [9, 18, 27] (0-indexed) and concatenates
cake-core/src/models/flux/text_encoder.rs:378
↓ 4 callersMethodencode_dialog_to_prompt
(&self)
cake-core/src/models/gemma3/history.rs:41
↓ 4 callersFunctionexaone4_config
()
cake-core/tests/unit_tests/test_blocks.rs:327
↓ 4 callersMethodexp_mul
(&self, x: &Tensor, y: &Tensor)
cake-core/src/backends/cpu/mod.rs:256
↓ 4 callersMethodforward
(&self, ids: &Tensor)
cake-core/src/models/flux/flux1_model.rs:383
↓ 4 callersMethodforward
(&self, src_orig: &Tensor, processed: &Tensor)
cake-core/src/models/luxtts/model.rs:43
↓ 4 callersMethodforward
(&self, t: &Tensor)
cake-core/src/models/vibevoice/prediction_head.rs:29
↓ 4 callersMethodget
( &self, s: Shape, name: &str, _h: Init, dtype: DType, dev: &D
cake-core/src/utils/native_dtype_backend.rs:54
↓ 4 callersMethodget
( &self, s: Shape, name: &str, _h: candle_nn::Init, dtype: DType,
cake-core/src/models/vibevoice/vibevoice_1_5b.rs:22
↓ 4 callersFunctionget_broadcast_addresses
Get directed broadcast addresses for all local IPv4 interfaces. Falls back to 255.255.255.255 if enumeration fails.
cake-core/src/cake/sharding/discovery.rs:500
↓ 4 callersFunctioninject_kv_cache
Inject a pre-computed KV cache into a cake Cache. This sets the KV pairs for each layer so that subsequent forward passes continue from where the voic
cake-core/src/models/vibevoice/voice_prompt.rs:77
↓ 4 callersMethodload_tensor
Load a tensor by name and return it as a candle Tensor on the given device.
cake-core/src/utils/tensor_storage.rs:270
↓ 4 callersFunctionmake_test_sd_context
(version: StableDiffusionVersion)
cake-core/src/models/sd/util.rs:292
↓ 4 callersFunctionmake_test_tokens
()
cake-core/src/models/luxtts/tokenizer.rs:244
↓ 4 callersFunctionmake_vb_convolution
(dim: usize, kernel_size: usize)
cake-core/tests/unit_tests/test_luxtts.rs:59
↓ 4 callersFunctionmake_vb_qwen3_5_linear
Build VarBuilder for a Qwen3_5 linear attention (GatedDeltaNet) block.
cake-core/benches/bench_helpers.rs:466
↓ 4 callersFunctionmake_vb_qwen3_5_linear
Build VarBuilder for a Qwen3_5 linear attention (GatedDeltaNet) block.
cake-core/tests/unit_tests/test_blocks.rs:452
↓ 4 callersFunctionmoe_config
()
cake-core/tests/unit_tests/test_moe.rs:13
↓ 4 callersFunctionolmo2_config
()
cake-core/tests/unit_tests/test_blocks.rs:269
↓ 4 callersFunctionqwen3_moe_config
()
cake-core/tests/unit_tests/test_blocks.rs:704
↓ 4 callersMethodrelease_output
(&self, buf: MappedBuffer)
cake-core/src/backends/vulkan/mod.rs:541
↓ 4 callersMethodrepeat_kv
(&self, x: Tensor)
cake-core/src/models/qwen3_5/full_attention.rs:118
↓ 4 callersMethodrepo
(&self)
cake-core/src/lib.rs:448
↓ 4 callersMethodsub_mul
(&self, a: &Tensor, b: &Tensor, c: &Tensor)
cake-core/src/backends/cpu/mod.rs:260
↓ 4 callersMethodtensor_slice_bytes
(&self, name: &str, byte_offset: usize, byte_len: usize)
cake-core/src/utils/tensor_storage.rs:607
↓ 4 callersFunctiontest_backend
()
cake-core/src/models/flux/flux1_model.rs:76
↓ 4 callersMethodtime_schedule
Generate the time schedule: linspace(0, 1, num_steps+1) with t_shift applied.
cake-core/src/models/luxtts/euler_solver.rs:25
↓ 3 callersFunctionInfoRow
cake-mobile-app/shared/src/commonMain/kotlin/com/evilsocket/cake/ui/WorkerScreen.kt:261
↓ 3 callersFunctionapply_rope
(x: &Tensor, freq_cis: &Tensor)
cake-core/src/models/flux/flux1_model.rs:328
↓ 3 callersFunctionarch_from_gguf
Detect the architecture string from GGUF metadata and map to TextModelArch.
cake-core/src/utils/gguf.rs:158
↓ 3 callersMethodas_new
Return a copy of this cache with the same state but new kv table.
cake-core/src/models/common/cache.rs:241
↓ 3 callersFunctionattention
(q: &Tensor, k: &Tensor, v: &Tensor, pe: &Tensor, backend: &dyn ComputeBackend)
cake-core/src/models/flux/flux1_model.rs:339
↓ 3 callersFunctionconfig_from_gguf
Extract a `Config` from GGUF metadata.
cake-core/src/utils/gguf.rs:78
↓ 3 callersFunctionconstant_time_eq
Constant-time comparison of two byte slices.
cake-core/src/cake/sharding/auth.rs:40
↓ 3 callersMethodconv2d
( &self, x: &Tensor, weight: &Tensor, bias: Option<&Tensor>, padding:
cake-core/src/backends/mod.rs:658
↓ 3 callersMethodconv_transpose1d
( &self, x: &Tensor, weight: &Tensor, bias: Option<&Tensor>, padding:
cake-core/src/backends/mod.rs:633
↓ 3 callersMethodconvert_v_to_x0
Convert v-prediction to x0 prediction. x0 = alpha_t * sample - sigma_t * v_pred
cake-core/src/models/vibevoice/ddpm.rs:86
↓ 3 callersFunctioncpu_backend
()
cake-core/benches/bench_attention.rs:8
↓ 3 callersFunctioncpu_backend
()
cake-core/benches/bench_mlp.rs:8
↓ 3 callersMethoddecode
( forwarder: &mut Box<dyn Forwarder>, latents: Tensor, ctx: &mut Context, )
cake-core/src/models/flux/vae.rs:150
↓ 3 callersFunctiondequantize_fp8_blockwise
Dequantize a 2-D FP8 weight tensor using its per-block scale factor.
cake-core/src/utils/fp8.rs:43
↓ 3 callersMethoddevice
(&self)
cake-core/src/backends/rocm/mod.rs:357
↓ 3 callersFunctiondtype_to_u8
Map a candle DType to a compact u8 tag for wire encoding.
cake-core/src/cake/sharding/proto/message.rs:8
↓ 3 callersFunctionflash_attention
( q: &Tensor, k: &Tensor, v: &Tensor, softmax_scale: f32, causal: bool, )
cake-core/src/utils/flash_attn.rs:11
↓ 3 callersMethodflat_to_stack
Map a flat layer index to (stack_idx, layer_in_stack).
cake-core/src/models/luxtts/config.rs:103
↓ 3 callersMethodforward
(&self, x: &Tensor, backend: &dyn ComputeBackend)
cake-core/src/models/flux/flux2_vae.rs:29
↓ 3 callersMethodforward
Apply bypass: output = x + scale * (module_out - x) bypass_scale is the weight on the non-residual (processed) path
cake-core/src/models/luxtts/bypass_module.rs:35
↓ 3 callersMethodforward
(&self, x: &Tensor)
cake-core/src/models/vibevoice/vae_encoder.rs:59
↓ 3 callersMethodforward
( &self, x: &Tensor, index_pos: usize, block_idx: usize, ctx: &mut Con
cake-core/src/models/common/transformer.rs:103
↓ 3 callersMethodforward_mut
( &mut self, x: &Tensor, index_pos: usize, block_idx: usize, ctx: &mut
cake-core/src/models/flux/vae.rs:97
↓ 3 callersMethodgenerate_text
Start the generation loop and call the stream function for every token. `max_tokens` overrides the default sample length if provided.
cake-core/src/cake/sharding/master.rs:109
↓ 3 callersMethodgetString
cake-mobile-app/shared/src/commonMain/kotlin/com/evilsocket/cake/Settings.kt:4
↓ 3 callersMethodget_conv_state
Get the conv state for a linear attention layer.
cake-core/src/models/common/cache.rs:231
↓ 3 callersFunctionget_inference_device
Returns the best available device at `ordinal` index (in case of multiple GPUs), or CPU if `force_cpu` is true.
cake-core/src/utils/mod.rs:144
↓ 3 callersMethodget_recurrent_state
Get the recurrent state for a linear attention layer.
cake-core/src/models/common/cache.rs:221
↓ 3 callersMethodgpu_alloc
(&self, count: usize)
cake-core/src/backends/rocm/mod.rs:209
↓ 3 callersMethodgpu_upload
(&self, data: &[f32])
cake-core/src/backends/rocm/mod.rs:217
↓ 3 callersMethodinto_config
Convert the LLM backbone config to cake's common Config.
cake-core/src/models/vibevoice/config.rs:82
↓ 3 callersMethodlayer_norm
(&self, x: &Tensor, weight: &Tensor, bias: Option<&Tensor>, eps: f32)
cake-core/src/backends/metal/mod.rs:927
↓ 3 callersFunctionlist_models
Scan all known locations for models and return a list of discovered models.
cake-core/src/utils/models.rs:70
↓ 3 callersMethodload_all
Load all tensors as a HashMap suitable for VarBuilder::from_tensors().
cake-core/src/utils/tensor_storage.rs:296
↓ 3 callersFunctionload_fp8_var_builder
Create a VarBuilder that transparently dequantizes FP8 weights. # Safety Inherits the mmap safety requirements from `MmapedSafetensors`.
cake-core/src/utils/fp8.rs:138
↓ 3 callersFunctionload_index
(data_path: &Path)
cake-core/src/utils/split.rs:55
↓ 3 callersMethodload_var_builder
( &self, filenames: &[PathBuf], dtype: DType, device: &Device, )
cake-core/src/utils/mod.rs:67
↓ 3 callersFunctionload_var_builder_for_local_layers
Create a VarBuilder that only loads safetensors shards needed for the given local layers. Shards containing only remote-worker tensors are excluded, r
cake-core/src/utils/mod.rs:272
↓ 3 callersFunctionload_wav_mono
Load a WAV file and return mono PCM f32 samples at the given target sample rate. Supports 16-bit integer and 32-bit float WAV files, mono or stereo.
cake-core/src/utils/wav.rs:74
↓ 3 callersFunctionlooks_like_hf_repo
Returns true if the string looks like a HuggingFace repo ID (e.g., "evilsocket/Qwen2.5-Coder-1.5B-Instruct").
cake-core/src/utils/hf.rs:10
↓ 3 callersFunctionmake_f16_tensor
(shape: &[usize])
cake-core/tests/unit_tests/test_client_worker.rs:15
↓ 3 callersFunctionmake_f32_tensor
(shape: &[usize])
cake-core/src/cake/sharding/proto/message.rs:402
↓ 3 callersFunctionmake_mock_storage
(prefix: &str, num_experts: usize, i: usize, h: usize)
cake-core/src/models/common/disk_expert_provider.rs:729
↓ 3 callersFunctionmake_topology
(num_layers: usize)
cake-core/benches/bench_topology.rs:4
↓ 3 callersFunctionmake_vb_attention
Build VarBuilder for CausalSelfAttention.
cake-core/benches/bench_helpers.rs:135
↓ 3 callersFunctionmake_vb_feedforward
(dim: usize, ff_dim: usize)
cake-core/tests/unit_tests/test_luxtts.rs:10
↓ 3 callersFunctionmake_vb_mlp
Build VarBuilder for MLP.
cake-core/benches/bench_helpers.rs:179
↓ 3 callersFunctionmake_vb_qwen3_5_full
Build VarBuilder for a Qwen3_5 full attention block.
cake-core/tests/unit_tests/test_blocks.rs:585
↓ 3 callersMethodmap_ipa_to_tokens
Map an IPA string to token IDs by greedily matching the longest token.
cake-core/src/models/luxtts/tokenizer.rs:133
↓ 3 callersMethodmax_layers_for_size
Maximum number of layers this worker can fit, based on per-GPU VRAM.
cake-core/src/cake/sharding/discovery.rs:49
↓ 3 callersMethodname
Component name for topology layer assignment.
cake-core/src/models/flux/config.rs:76
↓ 3 callersFunctionprefetch_safetensors
Pre-read safetensor files into the OS page cache so that subsequent mmap access doesn't trigger per-tensor page faults during layer loading. Uses Once
cake-core/src/utils/mod.rs:213
↓ 3 callersFunctionprint_bench
(label: &str, size_bytes: usize, duration: Duration, iterations: usize)
cake-core/tests/protocol.rs:369
← previousnext →201–300 of 2,320, ranked by callers