MCPcopy Create free account

hub / github.com/tdrussell/diffusion-pipe / functions

Functions1,157 in github.com/tdrussell/diffusion-pipe

⨍ Functions 1,157◇ Types & classes 222
Method__init__
( self, *, # enforce keyword arguments head_dim: int, len_h: int, len
models/cosmos_predict2_modeling.py:492
Method__init__
Args: interpolation (str): we curretly only support "crop", ideally when we need extrapolation capacity, we should adjust frequen
models/cosmos_predict2_modeling.py:622
Method__init__
(self, num_channels: int)
models/cosmos_predict2_modeling.py:675
Method__init__
(self, in_features: int, out_features: int, use_adaln_lora: bool = False)
models/cosmos_predict2_modeling.py:698
Method__init__
(self, num_channels: int, bandwidth: int = 1, normalize: bool = False)
models/cosmos_predict2_modeling.py:754
Method__init__
( self, spatial_patch_size: int, temporal_patch_size: int, in_channels: int =
models/cosmos_predict2_modeling.py:804
Method__init__
( self, hidden_size: int, spatial_patch_size: int, temporal_patch_size: int,
models/cosmos_predict2_modeling.py:864
Method__init__
( self, max_img_h: int, max_img_w: int, max_frames: int, in_channels:
models/cosmos_predict2_modeling.py:1180
Method__init__
(self, config, support_video=False, framerate=None, round_height=16, round_width=16, round_frames=4)
models/base.py:77
Method__init__
(self, *args, **kwargs)
models/base.py:178
Method__init__
(self, load_fn)
models/base.py:453
Method__init__
(self, config)
models/base.py:477
Method__init__
(self, z_channels: int, z_factor: int, embedding_dim: int, **kwargs)
models/cosmos.py:49
Method__init__
(self, config)
models/cosmos.py:141
Method__init__
(self, transformer, spatial_compression_factor)
models/cosmos.py:297
Method__init__
(self, block)
models/cosmos.py:373
Method__init__
(self, config)
models/sdxl.py:371
Method__init__
(self, diffusers_pipeline)
models/sdxl.py:653
Method__init__
(self, resnet, attn, append_residual_hidden_states=True)
models/sdxl.py:786
Method__init__
(self, resnet, attn)
models/sdxl.py:803
Method__init__
(self, resnet, attn)
models/sdxl.py:818
Method__init__
(self, downsamplers)
models/sdxl.py:836
Method__init__
(self, upsamplers, is_final_block)
models/sdxl.py:850
Method__init__
(self, block)
models/sdxl.py:868
Method__init__
(self, block)
models/sdxl.py:900
Method__init__
(self, block, is_final_block)
models/sdxl.py:930
Method__init__
(self, config)
models/ltx_video.py:31
Method__init__
(self, transformer)
models/ltx_video.py:222
Method__init__
(self, block)
models/ltx_video.py:279
Method__init__
(self, config)
models/cosmos_predict2.py:185
Method__init__
(self, model, text_encoder, is_generic_llm)
models/cosmos_predict2.py:533
Method__init__
(self, llm_adapter)
models/cosmos_predict2.py:578
Method__init__
(self, block, block_idx, offloader)
models/cosmos_predict2.py:599
Method__init__
(self, model)
models/cosmos_predict2.py:617
Method__init__
(self, config)
models/hunyuan_video.py:196
Method__init__
(self, transformer)
models/hunyuan_video.py:543
Method__init__
(self, block, block_idx, offloader)
models/hunyuan_video.py:645
Method__init__
(self, transformer)
models/hunyuan_video.py:662
Method__init__
(self, head_dim)
models/llm_adapter.py:51
Method__init__
(self, query_dim, context_dim, n_heads, head_dim)
models/llm_adapter.py:73
Method__init__
(self, source_dim, model_dim, num_heads=16, mlp_ratio=4.0, self_attn=False, layer_norm=False)
models/llm_adapter.py:118
Method__init__
(self, source_dim, target_dim, model_dim, num_layers=6, num_heads=16, embed=None, self_attn=False, layer_norm=
models/llm_adapter.py:165
Method__init__
(self, config)
models/sd3.py:30
Method__init__
(self, model)
models/sd3.py:171
Method__init__
(self, block)
models/sd3.py:199
Method__init__
(self, config)
models/ltx2.py:197
Method__init__
(self, model, framerate)
models/ltx2.py:398
Method__init__
(self, layer, block_idx, offloader, split_mode)
models/ltx2.py:554
Method__init__
(self, config)
models/hunyuan_image.py:54
Method__init__
(self, transformer)
models/hunyuan_image.py:385
Method__init__
(self, block, block_idx, offloader)
models/hunyuan_image.py:481
Method__init__
(self, transformer)
models/hunyuan_image.py:498
Method__init__
(self, *args, **kwargs)
models/ideogram4.py:30
Method__init__
(self, model)
models/ideogram4.py:135
Method__init__
(self, layer, block_idx, offloader)
models/ideogram4.py:226
Method__init__
(self, model)
models/ideogram4.py:244
Method__init__
(self, *args, **kwargs)
models/krea2.py:27
Method__init__
(self, model)
models/krea2.py:182
Method__init__
(self, layer, block_idx, offloader)
models/krea2.py:248
Method__init__
(self, config)
models/omnigen2.py:39
Method__init__
(self, model)
models/omnigen2.py:191
Method__init__
(self, block)
models/omnigen2.py:290
Method__init__
(self, *args, **kwargs)
models/flux2.py:26
Method__init__
(self, model)
models/flux2.py:235
Method__init__
(self, layer, block_idx, offloader)
models/flux2.py:330
Method__init__
(self, model)
models/flux2.py:357
Method__init__
(self, layer, block_idx, offloader)
models/flux2.py:377
Method__init__
( self, hidden_size: int, heads_num: int, mlp_width_ratio: float = 4.0,
models/hunyuan_image_modeling.py:247
Method__init__
(self, x_embedder, time_text_embed, context_embedder, pos_embed)
models/flux.py:455
Method__init__
(self, block, block_idx, offloader)
models/flux.py:489
Method__init__
(self, block, block_idx, offloader)
models/flux.py:512
Method__init__
(self, norm_out, proj_out)
models/flux.py:535
Method__init__
(self)
models/qwen_image.py:85
Method__init__
(self, config)
models/qwen_image.py:190
Method__init__
(self, model)
models/qwen_image.py:516
Method__init__
(self, block, block_idx, offloader)
models/qwen_image.py:558
Method__init__
(self, config)
models/wan/wan.py:73
Method__init__
(self, model, text_encoder)
models/wan/wan.py:414
Method__init__
(self, block, block_idx, offloader)
models/wan/wan.py:514
Method__init__
(self, model)
models/wan/wan.py:532
Method__init__
(self, name, seq_len=None, clean=None, **kwargs)
models/wan/tokenizers.py:37
Method__init__
(self, dim, num_heads, causal=False, attn_
models/wan/clip.py:49
Method__init__
(self, dim, mlp_ratio, num_heads, post_nor
models/wan/clip.py:108
Method__init__
(self, dim, mlp_ratio, num_heads, activati
models/wan/clip.py:152
Method__init__
(self, image_size=224, patch_size=16, dim=768,
models/wan/clip.py:205
Method__init__
(self, **kwargs)
models/wan/clip.py:299
Method__init__
(self, embed_dim=1024, image_size=224, patch_size=14,
models/wan/clip.py:324
Method__init__
(self, dtype, device, checkpoint_path, tokenizer_path)
models/wan/clip.py:497
Method__init__
(self, dim, eps=1e-6)
models/wan/t5.py:49
Method__init__
(self, dim, dim_attn, num_heads, dropout=0.1)
models/wan/t5.py:65
Method__init__
(self, dim, dim_ffn, dropout=0.1)
models/wan/t5.py:119
Method__init__
(self, dim, dim_attn, dim_ffn, num_heads,
models/wan/t5.py:140
Method__init__
(self, dim, dim_attn, dim_ffn, num_heads,
models/wan/t5.py:174
Method__init__
(self, num_buckets, num_heads, bidirectional, max_dist=128)
models/wan/t5.py:217
Method__init__
(self, vocab, dim, dim_attn, dim_ffn,
models/wan/t5.py:263
Method__init__
(self, vocab, dim, dim_attn, dim_ffn,
models/wan/t5.py:311
Method__init__
( self, text_len, dtype=torch.bfloat16, device=torch.cuda.current_device(),
models/wan/t5.py:543
Method__init__
(self, *args, **kwargs)
models/wan/vae2_2.py:22
Method__init__
(self, dim, channel_first=True, images=True, bias=False)
models/wan/vae2_2.py:47
Method__init__
(self, dim, mode)
models/wan/vae2_2.py:73
← previousnext →501–600 of 1,157, ranked by callers