Crate diffusion_rs_common

Modules

• core
• nn
  candle-nn

Macros

• bail
• map_dtype
• test_device

Structs

• NiceProgressBar
  Nice progress bar with over an iterator and a message. COLOR is one of r,g,b

Enums

• FileData
• FileLoader
• ModelSource
  Source from which to load the model. This is easiest to create with the various constructor functions.
• TokenSource
  The source of the HF token.

Functions

• conv1d
• conv1d_no_bias
• conv2d
• conv2d_no_bias
• embedding
• from_mmaped_safetensors
  Load tensors into a VarBuilder backed by a VarMap using MmapedSafetensors. Set silent to not show a progress bar.
• get_token
  This reads a token from a specified source. If the token cannot be read, a warning is logged with tracing and no token is used.
• group_norm
• layer_norm
• linear
• linear_b
• linear_no_bias
• load_bpe_tokenizer

Type Aliases

• VarBuilder
  A simple VarBuilder, this is less generic than VarBuilderArgs but should cover most common use cases.