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

↓ 1 callersFunctiontimestep_embedding

( timesteps: Tensor<B, 1, Int>, dim: usize, max_period: usize, )

src/model/unet/mod.rs:19

↓ 1 callersMethodunconditional_context

(&self, tokenizer: &SimpleTokenizer)

src/model/stablediffusion/mod.rs:194

↓ 1 callersFunctionwhitespace_clean

(text)

python/dump.py:472

↓ 1 callersFunctionwhitespace_clean

(text)

python/tokenizer.py:56

↓ 1 callersFunctionwhitespace_clean

(self, x, context=None)

(self, x, context=None)

python/dump.py:232

Method__call__

(self, x, context=None)

(self, x, timesteps=None, context=None)

python/dump.py:326

Method__call__

(self, hidden_states)

python/dump.py:357

Method__call__

(self, hidden_states, causal_attention_mask)

python/dump.py:377

Method__call__

(self, hidden_states, causal_attention_mask)

python/dump.py:413

Method__call__

(self, hidden_states, causal_attention_mask)

python/dump.py:430

Method__call__

(self, input_ids, position_ids)

(self, in_channels, out_channels=None)

(self, channels, emb_channels, out_channels)

python/dump.py:153

Method__init__

(self, query_dim, context_dim, n_heads, d_head)

python/dump.py:180

Method__init__

(self, dim_in, dim_out)

(self, dim, context_dim, n_heads, d_head)

python/dump.py:224

Method__init__

(self, channels, context_dim, n_heads, d_head)

(self, bpe_path: str = default_bpe())

(self, bpe_path: str = default_bpe())

python/tokenizer.py:63

Functionattn_decoder_mask

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

src/backend.rs:130

Functionconvert_dump_to_model

( dump_path: &str, model_name: &str, device: &B::Device, )

src/bin/convert/main.rs:18

Functioncosine_schedule

(n_steps: i64, device: &B::Device)

src/model/stablediffusion/mod.rs:216

Methoddecode

(&self, tokens: &[i64], skip_special_tokens: bool)

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

src/model/silu.rs:14

Methodforward

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

src/model/groupnorm/mod.rs:53

Methodforward

( &self, x: Tensor<B, 4>, timesteps: Tensor<B, 1, Int>, context: Tensor<B, 3>,

src/model/unet/mod.rs:109

Methodinit

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

src/model/groupnorm/mod.rs:18

Methodinit

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

src/model/stablediffusion/mod.rs:23

Methodinit

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

src/model/unet/mod.rs:36

Functionload_decoder_block

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

src/model/autoencoder/load.rs:103

Functionload_downsample

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

src/model/unet/load.rs:139

Functionload_encoder_block

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

src/model/autoencoder/load.rs:120

Functionload_f32

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

src/model/load.rs:49

Functionload_geglu

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

src/model/unet/load.rs:63

Functionload_mlp

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

src/model/unet/load.rs:74

Functionload_multi_head_attention

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

src/model/unet/load.rs:42

Functionload_res_block

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

src/model/unet/load.rs:16

Functionload_res_transformer

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

src/model/unet/load.rs:197

Functionload_res_transformer_res

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

src/model/unet/load.rs:146

Functionload_res_transformer_upsample

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

src/model/unet/load.rs:164

Functionload_res_upsample

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

src/model/unet/load.rs:182

Functionload_residual_decoder_attention_block

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

src/model/clip/load.rs:49

Functionload_spatial_transformer

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

src/model/unet/load.rs:109

Functionload_transformer_block

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

src/model/unet/load.rs:86

Functionload_unet_input_blocks

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

src/model/unet/load.rs:213

Functionload_unet_output_blocks

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

src/model/unet/load.rs:248

Functionload_upsample

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

src/model/unet/load.rs:128

Functionload_usize

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

src/model/load.rs:57

Functionmain

()

src/bin/sample/main.rs:36

Functionmain

()

src/bin/convert/main.rs:39

(tokenizer_path: &str)

src/model/clip/mod.rs:220

Functionoffset_cosine_schedule

(n_steps: i64, device: &B::Device)

src/model/stablediffusion/mod.rs:223

Functionoffset_cosine_schedule_cumprod

(n_steps: i64, device: &B::Device)

src/model/stablediffusion/mod.rs:235

Functionqkv_attention

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

Functionsave_stable_diffusion

(stable_diffusion, path)

python/stablediffusion.py:8

Functionsave_test_image

save red test image

src/bin/sample/main.rs:128

Functiontest_encode_decode

()

src/tokenizer.rs:210

Methodvocab_size

(&self, include_special_tokens: bool)

src/token.rs:46

← previous101–194 of 194, ranked by callers