Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/tdrussell/diffusion-pipe
/ functions
Functions
1,157 in github.com/tdrussell/diffusion-pipe
⨍
Functions
1,157
◇
Types & classes
222
Method
forward
(self, inputs)
models/flux2.py:365
Method
forward
(self, inputs)
models/flux2.py:384
Method
forward
(self, inputs)
models/flux2.py:410
Method
forward
( self, img: torch.Tensor, txt: torch.Tensor, vec: torch.Tensor, freqs
models/hunyuan_image_modeling.py:149
Method
forward
( self, x: torch.Tensor, vec: torch.Tensor, txt_len: int, freqs_cis: T
models/hunyuan_image_modeling.py:301
Method
forward
(self, inputs)
models/flux.py:463
Method
forward
(self, inputs)
models/flux.py:496
Method
forward
(self, inputs)
models/flux.py:519
Method
forward
(self, inputs)
models/flux.py:541
Method
forward
(self, inputs)
models/qwen_image.py:525
Method
forward
(self, inputs)
models/qwen_image.py:565
Method
forward
(self, inputs)
models/qwen_image.py:593
Method
forward
(self, inputs)
models/wan/wan.py:432
Method
forward
(self, inputs)
models/wan/wan.py:521
Method
forward
(self, inputs)
models/wan/wan.py:541
Method
forward
(self, x)
models/wan/clip.py:37
Method
forward
(self, x)
models/wan/clip.py:43
Method
forward
x: [B, L, C].
models/wan/clip.py:68
Method
forward
(self, x)
models/wan/clip.py:140
Method
forward
x: [B, L, C].
models/wan/clip.py:180
Method
forward
(self, x, interpolation=False, use_31_block=False)
models/wan/clip.py:273
Method
forward
(self, ids)
models/wan/clip.py:309
Method
forward
imgs: [B, 3, H, W] of torch.float32. - mean: [0.48145466, 0.4578275, 0.40821073] - std: [0.26862954, 0.2613025
models/wan/clip.py:400
Method
forward
(self, x)
models/wan/t5.py:42
Method
forward
(self, x)
models/wan/t5.py:55
Method
forward
x: [B, L1, C]. context: [B, L2, C] or None. mask: [B, L2] or [B, L1, L2] or None.
models/wan/t5.py:80
Method
forward
(self, x)
models/wan/t5.py:130
Method
forward
(self, x, mask=None, pos_bias=None)
models/wan/t5.py:164
Method
forward
(self, x, mask=None, encoder_states=None, enco
models/wan/t5.py:200
Method
forward
(self, lq, lk)
models/wan/t5.py:227
Method
forward
(self, ids, mask=None)
models/wan/t5.py:297
Method
forward
(self, ids, mask=None, encoder_states=None, encoder_mask=None)
models/wan/t5.py:345
Method
forward
(self, encoder_ids, encoder_mask, decoder_ids, decoder_mask)
models/wan/t5.py:402
Method
forward
(self, x, cache_x=None)
models/wan/vae2_2.py:34
Method
forward
(self, x)
models/wan/vae2_2.py:57
Method
forward
Fix bfloat16 support for nearest neighbor interpolation.
models/wan/vae2_2.py:64
Method
forward
(self, x, feat_cache=None, feat_idx=[0])
models/wan/vae2_2.py:112
Method
forward
(self, x, feat_cache=None, feat_idx=[0])
models/wan/vae2_2.py:214
Method
forward
(self, x)
models/wan/vae2_2.py:255
Method
forward
(self, x: torch.Tensor)
models/wan/vae2_2.py:335
Method
forward
(self, x, feat_cache=None, feat_idx=[0])
models/wan/vae2_2.py:447
Method
forward
(self, x, feat_cache=None, feat_idx=[0], first_chunk=False)
models/wan/vae2_2.py:489
Method
forward
(self, x, feat_cache=None, feat_idx=[0])
models/wan/vae2_2.py:559
Method
forward
(self, x, feat_cache=None, feat_idx=[0], first_chunk=False)
models/wan/vae2_2.py:672
Method
forward
(self, x, scale=[0, 1])
models/wan/vae2_2.py:778
Method
forward
(self, x, cache_x=None)
models/wan/vae2_1.py:24
Method
forward
(self, x)
models/wan/vae2_1.py:47
Method
forward
Fix bfloat16 support for nearest neighbor interpolation.
models/wan/vae2_1.py:55
Method
forward
(self, x, feat_cache=None, feat_idx=[0])
models/wan/vae2_1.py:97
Method
forward
(self, x, feat_cache=None, feat_idx=[0])
models/wan/vae2_1.py:198
Method
forward
(self, x)
models/wan/vae2_1.py:236
Method
forward
(self, x, feat_cache=None, feat_idx=[0])
models/wan/vae2_1.py:314
Method
forward
(self, x, feat_cache=None, feat_idx=[0])
models/wan/vae2_1.py:419
Method
forward
x: [B, L, C].
models/wan/xlm_roberta.py:25
Method
forward
(self, x, mask)
models/wan/xlm_roberta.py:64
Method
forward
ids: [B, L] of torch.LongTensor.
models/wan/xlm_roberta.py:116
Method
forward
r""" Args: x(Tensor): Shape [B, L, C]
models/wan/model.py:78
Method
forward
r""" Args: x(Tensor): Shape [B, L, C]
models/wan/model.py:94
Method
forward
r""" Args: x(Tensor): Shape [B, L, num_heads, C / num_heads] seq_lens(Tensor): Shape [B] grid_sizes(Tensor
models/wan/model.py:127
Method
forward
r""" Args: x(Tensor): Shape [B, L1, C] context(Tensor): Shape [B, L2, C] context_lens(Tensor): Shape [B]
models/wan/model.py:161
Method
forward
r""" Args: x(Tensor): Shape [B, L1, C] context(Tensor): Shape [B, L2, C] context_lens(Tensor): Shape [B]
models/wan/model.py:199
Method
forward
r""" Args: x(Tensor): Shape [B, L, C] e(Tensor): Shape [B, L1, 6, C] seq_lens(Tensor): Shape [B], length o
models/wan/model.py:277
Method
forward
(self, image_embeds)
models/wan/model.py:359
Method
from_weakref
(cls, cdata)
utils/reduction.py:43
Method
generate_embeddings
Generate embeddings for the given input size. Args: B_T_H_W_C (torch.Size): Input tensor size (Batch, Time, Height, Widt
models/cosmos_predict2_modeling.py:549
Method
generate_embeddings
(self, B_T_H_W_C: torch.Size, fps: Optional[torch.Tensor])
models/cosmos_predict2_modeling.py:654
Method
get_avg_learning_rate
(self)
optimizers/automagic.py:173
Method
get_call_text_encoder_fn
(self, text_encoder)
models/hidream.py:147
Method
get_call_text_encoder_fn
(self, text_encoder)
models/chroma.py:194
Method
get_call_text_encoder_fn
(self, text_encoder)
models/hunyuan_video_15.py:31
Method
get_call_text_encoder_fn
(self, text_encoder)
models/auraflow.py:98
Method
get_call_text_encoder_fn
(self, text_encoder)
models/lumina_2.py:113
Method
get_call_text_encoder_fn
(self, text_encoder)
models/base.py:645
Method
get_call_text_encoder_fn
(self, text_encoder)
models/cosmos.py:212
Method
get_call_text_encoder_fn
(self, text_encoder)
models/ltx_video.py:108
Method
get_call_text_encoder_fn
(self, text_encoder)
models/cosmos_predict2.py:341
Method
get_call_text_encoder_fn
(self, text_encoder)
models/hunyuan_video.py:364
Method
get_call_text_encoder_fn
(self, text_encoder)
models/hunyuan_image.py:244
Method
get_call_text_encoder_fn
(self, text_encoder)
models/omnigen2.py:126
Method
get_call_text_encoder_fn
(self, text_encoder)
models/flux.py:305
Method
get_call_text_encoder_fn
(self, text_encoder)
models/qwen_image.py:382
Method
get_call_text_encoder_fn
(self, text_encoder)
models/wan/wan.py:318
Method
get_call_vae_fn
(self, vae)
models/hidream.py:138
Method
get_call_vae_fn
(self, vae)
models/chroma.py:185
Method
get_call_vae_fn
(self, vae)
models/auraflow.py:87
Method
get_call_vae_fn
(self, vae)
models/lumina_2.py:104
Method
get_call_vae_fn
(self, vae)
models/base.py:209
Method
get_call_vae_fn
(self, vae)
models/cosmos.py:207
Method
get_call_vae_fn
(self, vae)
models/sdxl.py:527
Method
get_call_vae_fn
(self, vae)
models/ltx_video.py:98
Method
get_call_vae_fn
(self, vae)
models/cosmos_predict2.py:333
Method
get_call_vae_fn
(self, vae)
models/hunyuan_video.py:359
Method
get_call_vae_fn
(self, vae)
models/hunyuan_image.py:164
Method
get_call_vae_fn
(self, vae)
models/omnigen2.py:73
Method
get_call_vae_fn
Returns a function to encode images with the VAE, supporting optional control images. Similar to Qwen Edit implementation for referen
models/flux2.py:81
Method
get_call_vae_fn
(self, vae)
models/flux.py:290
Method
get_call_vae_fn
(self, vae)
models/qwen_image.py:296
Method
get_call_vae_fn
(self, vae_and_clip)
models/wan/wan.py:280
Method
get_emb_and_mask
(prompt_embeds)
models/hunyuan_image.py:264
Method
get_idxs
(self)
optimizers/projectors/topk_norm_projector.py:54
← previous
next →
801–900 of 1,157, ranked by callers