Functions — github.com/Gadersd/stable-diffusion-burn

hub / github.com/Gadersd/stable-diffusion-burn / functions

Functions194 in github.com/Gadersd/stable-diffusion-burn

⨍ Functions 194◇ Types & classes 88

↓ 24 callersFunctionsave_conv2d

(conv2d, path)

python/save.py:52

↓ 16 callersFunctionload_conv2d

( path: &str, device: &B::Device, )

src/model/load.rs:118

↓ 16 callersFunctionsave_tensor

(tensor, name, path)

python/save.py:10

↓ 15 callersFunctionsave_linear

(linear, path)

python/save.py:17

↓ 15 callersFunctionsave_scalar

(s, name, path)

python/save.py:6

↓ 13 callersFunctionsave_res_transformer

(block, path)

python/unet.py:106

↓ 9 callersFunctionsave_group_norm

(layer_norm, path)

python/save.py:29

↓ 9 callersFunctionsave_res_block

(res_block, path)

python/unet.py:8

↓ 7 callersFunctionload_resnet_block

( path: &str, device: &B::Device, )

src/model/autoencoder/load.rs:35

↓ 7 callersFunctionsave_resnet_block

(resnet_block, path)

python/autoencoder.py:7

↓ 6 callersFunctionload_linear

( path: &str, device: &B::Device, )

src/model/load.rs:65

↓ 6 callersFunctionsave_layer_norm

(layer_norm, path)

python/save.py:23

↓ 5 callersFunctionload_group_norm

( path: &str, device: &B::Device, )

src/model/groupnorm/load.rs:13

↓ 5 callersFunctiontensor_to_array_2

(x: Tensor<B, 1>)

src/model/load.rs:162

↓ 5 callersFunctionto_tuple_tensor

(val)

python/save.py:39

↓ 4 callersMethoddecode

(self, tokens)

python/tokenizer.py:129

↓ 4 callersMethodforward

(&self, x: Tensor<B, 4>)

src/model/autoencoder/mod.rs:56

↓ 3 callersMethod_shape

(self, tensor, seq_len: int, bsz: int)

python/dump.py:374

↓ 3 callersMethodforward

(&self, x: Tensor<B, 2, Int>)

src/model/clip/mod.rs:56

↓ 3 callersMethodinit

(&self, device: &B::Device)

src/model/clip/mod.rs:28

↓ 3 callersFunctionload_layer_norm

( path: &str, device: &B::Device, )

src/model/load.rs:89

↓ 3 callersFunctionqkv_attention

( q: Tensor<B, 3>, k: Tensor<B, 3>, v: Tensor<B, 3>, mask: Option<Tensor<B, 2>>, n_head: u

src/model/attention.rs:5

↓ 3 callersFunctionsave_downsample

(downsample, path)

python/unet.py:84

↓ 3 callersFunctionsave_spatial_transformer

(spatial_transformer, path)

python/unet.py:74

↓ 2 callersMethodas_array

(&self)

src/model/unet/mod.rs:162

↓ 2 callersFunctionbytes_to_unicode

Returns list of utf-8 byte and a corresponding list of unicode strings. The reversible bpe codes work on unicode strings. This means you need a

python/dump.py:477

↓ 2 callersFunctionbytes_to_unicode

Returns list of utf-8 byte and a corresponding list of unicode strings. The reversible bpe codes work on unicode strings. This means you

python/tokenizer.py:16

↓ 2 callersMethodcontext

(&self, tokenizer: &SimpleTokenizer, text: &str)

src/model/stablediffusion/mod.rs:198

↓ 2 callersMethoddecode_latent

(&self, latent: Tensor<B, 4>)

src/model/autoencoder/mod.rs:68

↓ 2 callersFunctionget_pairs

Return set of symbol pairs in a word. Word is represented as tuple of symbols (symbols being variable-length strings).

python/dump.py:461

↓ 2 callersFunctionget_pairs

Return set of symbol pairs in a word. Word is represented as tuple of symbols (symbols being variable-length strings).

python/tokenizer.py:38

↓ 2 callersFunctionget_pairs

(word: &[String])

src/tokenizer.rs:30

↓ 2 callersFunctionload_mid

(path: &str, device: &B::Device)

src/model/autoencoder/load.rs:58

↓ 2 callersFunctionload_stable_diffusion

( path: &str, device: &B::Device, )

src/model/stablediffusion/load.rs:16

↓ 2 callersFunctionsave_cross_attention

(cross_attention, path)

python/unet.py:32

↓ 2 callersFunctionsave_embedding

(embedding, path)

python/save.py:97

↓ 2 callersFunctionsave_mid

(mid, path)

python/autoencoder.py:27

↓ 2 callersFunctionsave_res_transformer_upsample

(block, path)

python/unet.py:110

↓ 2 callersFunctionsave_upsample

(upsample, path)

python/unet.py:90

↓ 1 callersFunctionattn_decoder_mask

(seq_length: usize, device: &B::Device)

src/model/attention.rs:47

↓ 1 callersFunctionbasic_clean

(text)

python/tokenizer.py:50

python/tokenizer.py:80

↓ 1 callersMethodbpe

(&self, token: &str)

src/tokenizer.rs:118

↓ 1 callersFunctionbytes_to_unicode

()

src/tokenizer.rs:7

↓ 1 callersFunctionconstruct_vocab

( chars: impl Iterator<Item = char> + Clone, merges: &[(String, String)], )

src/tokenizer.rs:59

↓ 1 callersMethoddecode

(&self, tokens: &[u32])

src/tokenizer.rs:197

↓ 1 callersFunctiondefault_bpe

()

python/dump.py:458

↓ 1 callersFunctiondefault_bpe

()

python/tokenizer.py:11

↓ 1 callersFunctiondiv_roundup

(x: usize, y: usize)

src/model/autoencoder/mod.rs:376

↓ 1 callersMethodencode

(self, text)

python/tokenizer.py:121

↓ 1 callersMethodencode

(&self, text: &str)

src/tokenizer.rs:175

↓ 1 callersMethodencode

(&self, text: &str, include_bos: bool, include_eos: bool)

src/token.rs:20

↓ 1 callersMethodencode_image

(&self, x: Tensor<B, 4>)

src/model/autoencoder/mod.rs:60

↓ 1 callersMethodforward_diffuser

( &self, latent: Tensor<B, 4>, timestep: Tensor<B, 1, Int>, context: Tensor<B,

src/model/stablediffusion/mod.rs:162

↓ 1 callersMethodinit

(&self, device: &B::Device)

src/model/autoencoder/mod.rs:30

↓ 1 callersMethodlatent_to_image

(&self, latent: Tensor<B, 4>)

src/model/stablediffusion/mod.rs:69

↓ 1 callersFunctionlayernorm

(x: Tensor<B, D>, eps: f64)

src/model/groupnorm/mod.rs:75

↓ 1 callersFunctionload_autoencoder

( path: &str, device: &B::Device, )

src/model/autoencoder/load.rs:183

↓ 1 callersFunctionload_clip

(path: &str, device: &B::Device)

src/model/clip/load.rs:68

↓ 1 callersFunctionload_conv_self_attention_block

( path: &str, device: &B::Device, )

src/model/autoencoder/load.rs:16

↓ 1 callersFunctionload_decoder

(path: &str, device: &B::Device)

src/model/autoencoder/load.rs:135

↓ 1 callersFunctionload_embedding

( path: &str, device: &B::Device, )

src/model/load.rs:78

↓ 1 callersFunctionload_encoder

(path: &str, device: &B::Device)

src/model/autoencoder/load.rs:159

↓ 1 callersFunctionload_merges

(path: &str)

src/tokenizer.rs:41

↓ 1 callersFunctionload_mlp

(path: &str, device: &B::Device)

src/model/clip/load.rs:14

↓ 1 callersFunctionload_multi_head_self_attention

( path: &str, device: &B::Device, )

src/model/clip/load.rs:28

↓ 1 callersFunctionload_padded_conv2d

( path: &str, device: &B::Device, )

src/model/autoencoder/load.rs:70

↓ 1 callersFunctionload_stable_diffusion_model_file

( filename: &str, device: &B::Device, )

src/bin/sample/main.rs:27

↓ 1 callersFunctionload_tensor

( name: &str, path: &str, device: &B::Device, )

src/model/load.rs:30

↓ 1 callersFunctionload_unet

(path: &str, device: &B::Device)

src/model/unet/load.rs:281

↓ 1 callersFunctionnumpy_to_tensor

( numpy_data: NpyData<f32>, device: &B::Device, )

src/model/load.rs:17

↓ 1 callersMethodsample_image

( &self, context: Tensor<B, 3>, unconditional_context: Tensor<B, 2>, unconditi

src/model/stablediffusion/mod.rs:51

↓ 1 callersMethodsample_latent

( &self, context: Tensor<B, 3>, unconditional_context: Tensor<B, 2>, unconditi

src/model/stablediffusion/mod.rs:102

↓ 1 callersFunctionsave_attn_block

(attn_block, path)

python/autoencoder.py:18

↓ 1 callersFunctionsave_autoencoder

(autoencoder, path)

python/autoencoder.py:86

↓ 1 callersFunctionsave_basic_transformer_block

(basic_transformer_block, path)

python/unet.py:60

↓ 1 callersFunctionsave_clip_attention

(clip_attention, path)

python/clip.py:10

↓ 1 callersFunctionsave_clip_encoder

(clip_encoder, path)

python/clip.py:25

↓ 1 callersFunctionsave_clip_encoder_layer

(clip_encoder_layer, path)

python/clip.py:18

↓ 1 callersFunctionsave_clip_text_embeddings

(clip_text_embeddings, path)

python/clip.py:31

↓ 1 callersFunctionsave_clip_text_transformer

(clip_text_transformer, path)

python/clip.py:36

↓ 1 callersFunctionsave_clipmlp

(clip_mlp, path)

python/clip.py:5

↓ 1 callersFunctionsave_decoder

(decoder, path)

python/autoencoder.py:46

↓ 1 callersFunctionsave_decoder_block

(decoder_block, path)

python/autoencoder.py:34

↓ 1 callersFunctionsave_encoder

(encoder, path)

python/autoencoder.py:72

↓ 1 callersFunctionsave_encoder_block

(encoder_block, path)

python/autoencoder.py:62

↓ 1 callersFunctionsave_feed_forward

(feed_forward, path)

python/unet.py:50

↓ 1 callersFunctionsave_geglu

(geglu, path)

python/unet.py:44

↓ 1 callersFunctionsave_images

(images: &Vec<Vec<u8>>, basepath: &str, width: u32, height: u32)

src/bin/sample/main.rs:118

↓ 1 callersFunctionsave_model_file

( model: StableDiffusion<B>, name: &str, )

src/bin/convert/main.rs:32

↓ 1 callersFunctionsave_padded_conv2d

(padded_conv2d, path)

python/save.py:70

↓ 1 callersFunctionsave_res_transformer_res

(block, path)

python/unet.py:97

↓ 1 callersFunctionsave_res_upsample

(block, path)

python/unet.py:102

↓ 1 callersFunctionsave_unet_input_blocks

(input_blocks, path)

python/unet.py:116

↓ 1 callersFunctionsave_unet_model

(model, path)

python/unet.py:144

↓ 1 callersFunctionsave_unet_output_blocks

(output_blocks, path)

python/unet.py:130

↓ 1 callersFunctiontensor_to_array

(x: Tensor<B, 1>)

src/model/load.rs:168

↓ 1 callersFunctiontimestep_embedding

(timesteps, dim, max_period=10000)

python/test.py:31

↓ 1 callersFunctiontimestep_embedding

(timesteps, dim, max_period=10000)

python/dump.py:274

next →1–100 of 194, ranked by callers