Functions — github.com/tdrussell/diffusion-pipe

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

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

⨍ Functions 1,157◇ Types & classes 222

↓ 1 callersMethodload_adapter_weights

(self, adapter_path, fuse=False, fuse_weight=1.0)

models/flux.py:236

↓ 1 callersMethodload_and_fuse_adapter

(self, path)

models/base.py:334

↓ 1 callersFunctionload_custom_video_vae

()

tools/cosmos_vae_test.py:82

↓ 1 callersFunctionload_custom_video_vae

(path)

models/cosmos.py:88

↓ 1 callersMethodload_diffusion_model

(self)

models/base.py:298

↓ 1 callersMethodload_image_for_vlm

(self, path)

models/qwen_image.py:310

↓ 1 callersFunctionload_state_dict

(args, pretrained_model_path)

models/hunyuan_video.py:84

↓ 1 callersFunctionmake_diffusers_to_bfl_map

(num_double_blocks: int = NUM_DOUBLE_BLOCKS, num_single_blocks: int = NUM_SINGLE_BLOCKS)

models/flux.py:90

↓ 1 callersMethodmodel_specific_dataset_config_validation

(self, dataset_config)

models/base.py:355

↓ 1 callersFunctionpatchify

(x, patch_size)

models/wan/vae2_2.py:280

↓ 1 callersMethodpatchify_and_embed

( self, x: torch.Tensor, cap_feats: torch.Tensor, cap_mask: torch.Tensor, t: torch.Tensor, transformer

models/z_image.py:160

↓ 1 callersMethodpatchify_and_embed

( self, x: List[torch.Tensor] | torch.Tensor, cap_feats: torch.Tensor, cap_mask: torch.Tensor, t: torc

models/lumina_2.py:221

↓ 1 callersFunctionpos_interpolate

(pos, seq_len)

models/wan/clip.py:16

↓ 1 callersMethodprepare_block_swap_inference

(self, disable_block_swap=False)

models/base.py:390

↓ 1 callersMethodprepare_block_swap_training

(self)

models/hidream.py:308

↓ 1 callersMethodprepare_block_swap_training

(self)

models/chroma.py:320

↓ 1 callersMethodprepare_block_swap_training

(self)

models/ernie_image.py:107

↓ 1 callersMethodprepare_block_swap_training

(self)

models/hunyuan_video_15.py:197

↓ 1 callersMethodprepare_block_swap_training

(self)

models/z_image.py:111

↓ 1 callersMethodprepare_block_swap_training

(self)

models/auraflow.py:175

↓ 1 callersMethodprepare_block_swap_training

(self)

models/base.py:387

↓ 1 callersMethodprepare_block_swap_training

(self)

models/cosmos_predict2.py:437

↓ 1 callersMethodprepare_block_swap_training

(self)

models/hunyuan_video.py:524

↓ 1 callersMethodprepare_block_swap_training

(self)

models/ltx2.py:365

↓ 1 callersMethodprepare_block_swap_training

(self)

models/hunyuan_image.py:366

↓ 1 callersMethodprepare_block_swap_training

(self)

models/ideogram4.py:122

↓ 1 callersMethodprepare_block_swap_training

(self)

models/krea2.py:169

↓ 1 callersMethodprepare_block_swap_training

(self)

models/flux2.py:216

↓ 1 callersMethodprepare_block_swap_training

(self)

models/flux.py:436

↓ 1 callersMethodprepare_block_swap_training

(self)

models/qwen_image.py:503

↓ 1 callersMethodprepare_block_swap_training

(self)

models/wan/wan.py:401

↓ 1 callersMethodprepare_inputs

(self, inputs, timestep_quantile=None)

models/sd3.py:107

↓ 1 callersMethodprepare_sample_test

(self, prompt, negative_prompt='', cfg=1)

models/base.py:188

↓ 1 callersFunctionprepare_scheduler_for_custom_training

(noise_scheduler)

models/sdxl.py:281

↓ 1 callersMethodpreprocess_text_embeds

(self, context, attention_mask)

models/ltx2.py:461

↓ 1 callersMethodprocess_epoch

(self, epoch, step, examples)

utils/saver.py:130

↓ 1 callersMethodprocess_step

(self, step, examples)

utils/saver.py:146

↓ 1 callersMethodproject

(self, full_rank_grad: torch.Tensor, n_iter)

optimizers/projectors/svd_projector.py:74

↓ 1 callersFunctionquantize_tensor

Quantize a floating-point tensor to the target dtype with appropriate scaling. Args: tensor: Input tensor (float) dtype: Tar

optimizers/optimizer_utils.py:65

↓ 1 callersFunctionrecursive_clone_tensors

(obj)

utils/dataset.py:130

↓ 1 callersFunctionreduce_nested_tensor

(nt)

utils/reduction.py:423

↓ 1 callersFunctionreduce_sparse_tensor

(sparse)

utils/reduction.py:481

↓ 1 callersMethodreparameterize

(self, mu, log_var)

models/wan/vae2_1.py:566

↓ 1 callersMethodreset

(self)

utils/dataset.py:1310

↓ 1 callersMethodreset_parameters

(self)

models/cosmos_predict2_modeling.py:537

↓ 1 callersMethodreset_parameters

(self)

models/cosmos_predict2_modeling.py:648

↓ 1 callersMethodreset_parameters

(self)

models/cosmos_predict2_modeling.py:764

↓ 1 callersFunctionreshape_weight_for_sd

(w)

models/sdxl.py:175

↓ 1 callersFunctionrotate_half

(x)

models/llm_adapter.py:37

↓ 1 callersFunctionround_down_to_multiple

(x, multiple)

utils/common.py:110

↓ 1 callersFunctionsample_t

(t, batch_size, quantile=None)

utils/common.py:155

↓ 1 callersMethodsave_adapter

(self, name)

utils/saver.py:58

↓ 1 callersMethodsave_full_model

(self, name)

utils/saver.py:87

↓ 1 callersFunctionset_config_defaults

(config)

train.py:93

↓ 1 callersMethodset_eval_quantile

(self, quantile)

utils/dataset.py:987

↓ 1 callersFunctionshuffle_caption

(caption: str, delimiter: str = ", ")

utils/dataset.py:51

↓ 1 callersFunctionshuffle_captions

(captions: list[str], count: int = 0, delimiter: str = ', ', caption_prefix: str = '')

utils/dataset.py:47

↓ 1 callersFunctionsinusoidal_embedding_1d

(dim, position)

models/wan/model.py:15

↓ 1 callersFunctionslice_t_distribution

(t, min_t=0.0, max_t=1.0)

utils/common.py:149

↓ 1 callersFunctionsplit_batch

(batch, pieces)

utils/dataset.py:1258

↓ 1 callersMethodstate_dict

(self)

utils/dataset.py:1402

↓ 1 callersMethodstep_hook

(self)

optimizers/automagic.py:152

↓ 1 callersMethodswap_paramiters

(self)

optimizers/automagic.py:101

↓ 1 callersMethodswap_weight_devices

(self, block_to_cpu: nn.Module, block_to_cuda: nn.Module)

utils/offloading.py:140

↓ 1 callersFunctionswap_weight_devices_cuda

(device: torch.device, layer_to_cpu: nn.Module, layer_to_cuda: nn.Module)

utils/offloading.py:43

↓ 1 callersFunctionswap_weight_devices_no_cuda

not tested

utils/offloading.py:89

↓ 1 callersFunctiontime_shift

(mu: float, sigma: float, t: torch.Tensor)

models/hidream.py:27

↓ 1 callersFunctiontime_shift

(mu: float, sigma: float, t: torch.Tensor)

models/chroma.py:26

↓ 1 callersFunctiontime_shift

(mu: float, sigma: float, t: torch.Tensor)

models/lumina_2.py:31

↓ 1 callersFunctiontime_shift

(mu: float, sigma: float, t: torch.Tensor)

models/cosmos_predict2.py:38

↓ 1 callersFunctiontime_shift

(mu: float, sigma: float, t: torch.Tensor)

models/sd3.py:15

↓ 1 callersFunctiontime_shift

(mu: float, sigma: float, t: torch.Tensor)

models/omnigen2.py:29

↓ 1 callersFunctiontime_shift

(mu: float, sigma: float, t: torch.Tensor)

models/flux.py:120

↓ 1 callersFunctionumt5_keys_mapping

(state_dict)

models/wan/t5.py:532

↓ 1 callersFunctionumt5_keys_mapping_comfy

(state_dict)

models/wan/t5.py:462

↓ 1 callersFunctionumt5_keys_mapping_kijai

(state_dict)

models/wan/t5.py:522

↓ 1 callersFunctionumt5_xxl

(**kwargs)

models/wan/t5.py:446

↓ 1 callersFunctionunbatch_iter

(batch)

utils/dataset.py:142

↓ 1 callersFunctionunpatchify

(x, patch_size)

models/wan/vae2_2.py:299

↓ 1 callersMethodunpatchify

(self, x_B_T_H_W_M: torch.Tensor)

models/cosmos_predict2_modeling.py:1411

↓ 1 callersMethodupdate_automagic_lr

(self, group, state, update, state_device)

optimizers/generic_optim.py:568

↓ 1 callersFunctionupdate_parameter

Updates a parameter tensor, handling both regular torch.Tensor and QBytesTensor cases with proper rescaling for quantized tensors. Args:

optimizers/optimizer_utils.py:93

↓ 1 callersFunctionvae_decode

(latents, vae)

tools/hunyuan_video_vae_test.py:32

↓ 1 callersMethodvae_decode

(self, latents)

models/base.py:601

↓ 1 callersFunctionvae_encode

(tensor, vae)

tools/hunyuan_video_vae_test.py:26

↓ 1 callersFunctionvae_encode

(tensor, vae)

models/cosmos.py:115

↓ 1 callersFunctionvae_encode

(tensor, vae)

models/cosmos_predict2.py:102

↓ 1 callersFunctionvae_encode

(tensor, vae)

models/hunyuan_video.py:184

↓ 1 callersMethodvae_encode

(self, img)

models/base.py:589

↓ 1 callersMethodvalidate

(self)

utils/dataset.py:523

↓ 1 callersMethodzero_grad

(self, set_to_none: bool = True)

train.py:69

Method__call__

(self, spec, mask_filepath, size_bucket=None)

models/base.py:96

Method__call__

( self, attn: Attention, hidden_states: torch.FloatTensor, # Image stream enc

models/qwen_image.py:91

Method__call__

(self, sequence, **kwargs)

models/wan/tokenizers.py:47

Method__call__

(self, texts, device)

models/wan/t5.py:582

Method__del__

(self)

utils/reduction.py:52

Method__del__

(self)

utils/offloading.py:226

utils/reduction.py:58

Method__getattr__

(self, name)

models/hidream.py:68

← previous next →301–400 of 1,157, ranked by callers