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
Function
optimizer_hook
(p)
train.py:741
Method
param_groups
(self)
optimizers/gradient_release.py:10
Method
param_groups
(self)
models/wan/clip.py:412
Function
polar_express_fn
Polar Express Sign Method: https://arxiv.org/pdf/2505.16932 by Noah Amsel, David Persson, Christopher Musco, Robert M. Gower.
optimizers/generic_optim.py:192
Method
post_init
(self, global_batch_size: dict, global_batch_size_image: dict, data_parallel_rank: int, data_parallel_world_si
utils/dataset.py:346
Method
prepare_block_swap_inference
(self, disable_block_swap=False)
models/hidream.py:316
Method
prepare_block_swap_inference
(self, disable_block_swap=False)
models/chroma.py:328
Method
prepare_block_swap_inference
(self, disable_block_swap=False)
models/ernie_image.py:112
Method
prepare_block_swap_inference
(self, disable_block_swap=False)
models/hunyuan_video_15.py:202
Method
prepare_block_swap_inference
(self, disable_block_swap=False)
models/z_image.py:116
Method
prepare_block_swap_inference
(self, disable_block_swap=False)
models/auraflow.py:180
Method
prepare_block_swap_inference
(self, disable_block_swap=False)
models/base.py:733
Method
prepare_block_swap_inference
(self, disable_block_swap=False)
models/cosmos_predict2.py:442
Method
prepare_block_swap_inference
(self, disable_block_swap=False)
models/hunyuan_video.py:532
Method
prepare_block_swap_inference
(self, disable_block_swap=False)
models/ltx2.py:370
Method
prepare_block_swap_inference
(self, disable_block_swap=False)
models/hunyuan_image.py:374
Method
prepare_block_swap_inference
(self, disable_block_swap=False)
models/ideogram4.py:127
Method
prepare_block_swap_inference
(self, disable_block_swap=False)
models/krea2.py:174
Method
prepare_block_swap_inference
(self, disable_block_swap=False)
models/flux2.py:224
Method
prepare_block_swap_inference
(self, disable_block_swap=False)
models/flux.py:444
Method
prepare_block_swap_inference
(self, disable_block_swap=False)
models/qwen_image.py:508
Method
prepare_block_swap_inference
(self, disable_block_swap=False)
models/wan/wan.py:406
Method
prepare_block_swap_training
(self)
models/base.py:730
Method
prepare_inputs
(self, inputs, timestep_quantile=None)
models/hidream.py:183
Method
prepare_inputs
(self, inputs, timestep_quantile=None)
models/chroma.py:221
Method
prepare_inputs
(self, inputs, timestep_quantile=None)
models/ernie_image.py:33
Method
prepare_inputs
(self, inputs, timestep_quantile=None)
models/hunyuan_video_15.py:108
Method
prepare_inputs
(self, inputs, timestep_quantile=None)
models/z_image.py:37
Method
prepare_inputs
(self, inputs, timestep_quantile=None)
models/auraflow.py:105
Method
prepare_inputs
(self, inputs, timestep_quantile=None)
models/lumina_2.py:137
Method
prepare_inputs
(self, inputs, timestep_quantile=None)
models/base.py:349
Method
prepare_inputs
(self, inputs, timestep_quantile=None)
models/base.py:692
Method
prepare_inputs
(self, inputs, timestep_quantile=None)
models/cosmos.py:240
Method
prepare_inputs
(self, inputs, timestep_quantile=None)
models/sdxl.py:536
Method
prepare_inputs
(self, inputs, timestep_quantile=None)
models/ltx_video.py:120
Method
prepare_inputs
(self, inputs, timestep_quantile=None)
models/cosmos_predict2.py:357
Method
prepare_inputs
(self, inputs, timestep_quantile=None)
models/hunyuan_video.py:411
Method
prepare_inputs
(self, inputs, timestep_quantile=None)
models/ltx2.py:290
Method
prepare_inputs
(self, inputs, timestep_quantile=None)
models/hunyuan_image.py:257
Method
prepare_inputs
(self, inputs, timestep_quantile=None)
models/ideogram4.py:61
Method
prepare_inputs
(self, inputs, timestep_quantile=None)
models/krea2.py:107
Method
prepare_inputs
(self, inputs, timestep_quantile=None)
models/omnigen2.py:137
Method
prepare_inputs
(self, inputs, timestep_quantile=None)
models/flux2.py:128
Method
prepare_inputs
(self, inputs, timestep_quantile=None)
models/flux.py:321
Method
prepare_inputs
(self, inputs, timestep_quantile=None)
models/qwen_image.py:390
Method
prepare_inputs
(self, inputs, timestep_quantile=None)
models/wan/wan.py:331
Function
print_model_info
(model)
train.py:150
Method
process_file
(file)
utils/dataset.py:635
Method
project
(self, full_rank_grad: torch.Tensor, n_iter)
optimizers/projectors/topk_norm_projector.py:32
Method
project
(self, full_rank_grad: torch.Tensor, n_iter)
optimizers/projectors/uniform_projector.py:18
Method
project_back
(self, low_rank_grad: torch.Tensor)
optimizers/projectors/topk_norm_projector.py:46
Method
project_back
(self, low_rank_grad: torch.Tensor)
optimizers/projectors/uniform_projector.py:31
Method
qkv_fn
(x)
models/wan/model.py:138
Function
rebuild_cuda_tensor
( tensor_cls, tensor_size, tensor_stride, tensor_offset, storage_cls, dtype, stora
utils/reduction.py:158
Function
rebuild_event
(device, handle)
utils/reduction.py:104
Function
rebuild_meta_tensor
( tensor_cls, tensor_size, tensor_stride, tensor_offset, dtype, storage_size_bytes,
utils/reduction.py:126
Function
rebuild_nested_tensor
( rebuild_buffer_func, rebuild_buffer_args, rebuild_sizes_func, rebuild_sizes_args, rebuil
utils/reduction.py:406
Function
rebuild_sparse_compressed_tensor
( rebuild_compressed_indices_func, rebuild_compressed_indices_args, rebuild_plain_indices_func,
utils/reduction.py:461
Function
rebuild_sparse_coo_tensor
( rebuild_indices_func, rebuild_indices_args, rebuild_values_func, rebuild_values_args, sh
utils/reduction.py:448
Function
rebuild_storage_empty
(cls)
utils/reduction.py:576
Function
rebuild_storage_fd
(cls, df, size)
utils/reduction.py:543
Function
rebuild_storage_filename
(cls, manager, handle, size, dtype=None)
utils/reduction.py:556
Function
rebuild_tensor
(cls, storage, metadata)
utils/reduction.py:113
Function
rebuild_typed_storage
(storage, dtype)
utils/reduction.py:580
Function
rebuild_typed_storage_child
(storage, storage_type)
utils/reduction.py:589
Function
reduce_event
(event)
utils/reduction.py:108
Function
reduce_storage
(storage)
utils/reduction.py:598
Function
reduce_typed_storage
(storage)
utils/reduction.py:585
Function
reduce_typed_storage_child
(storage)
utils/reduction.py:594
Method
reparameterize
(self, mu, log_var)
models/wan/vae2_2.py:841
Method
reset_parameters
(self)
models/cosmos_predict2_modeling.py:232
Method
rope_params
Args: index: [0, 1, 2, 3] 1D Tensor representing the position index of the token
models/qwen_image.py:546
Method
sample
(self, w=512, h=512)
models/base.py:263
Method
sample
(self, imgs, deterministic=False)
models/wan/vae2_1.py:571
Method
save_adapter
(self, save_dir, peft_state_dict)
models/hidream.py:132
Method
save_adapter
(self, save_dir, peft_state_dict)
models/chroma.py:176
Method
save_adapter
(self, save_dir, peft_state_dict)
models/auraflow.py:84
Method
save_adapter
(self, save_dir, peft_state_dict)
models/lumina_2.py:95
Method
save_adapter
(self, save_dir, peft_state_dict)
models/base.py:310
Method
save_adapter
(self, save_dir, sd)
models/base.py:609
Method
save_adapter
(self, save_dir, peft_state_dict)
models/cosmos.py:193
Method
save_adapter
(self, save_dir, peft_state_dict)
models/sdxl.py:465
Method
save_adapter
(self, save_dir, peft_state_dict)
models/ltx_video.py:79
Method
save_adapter
(self, save_dir, peft_state_dict)
models/cosmos_predict2.py:316
Method
save_adapter
(self, save_dir, peft_state_dict)
models/hunyuan_video.py:343
Method
save_adapter
(self, save_dir, peft_state_dict)
models/sd3.py:63
Method
save_adapter
(self, save_dir, peft_state_dict)
models/hunyuan_image.py:125
Method
save_adapter
(self, save_dir, peft_state_dict)
models/omnigen2.py:67
Method
save_adapter
(self, save_dir, peft_state_dict)
models/flux.py:229
Method
save_adapter
(self, save_dir, peft_state_dict)
models/qwen_image.py:284
Method
save_adapter
(self, save_dir, peft_state_dict)
models/wan/wan.py:258
Method
save_model
(self, save_dir, diffusers_sd)
models/chroma.py:182
Method
save_model
(self, save_dir, state_dict)
models/lumina_2.py:101
Method
save_model
(self, save_dir, diffusers_sd)
models/base.py:340
Method
save_model
(self, save_dir, sd)
models/base.py:639
Method
save_model
(self, save_dir, diffusers_sd)
models/sdxl.py:487
Method
save_model
(self, save_dir, diffusers_sd)
models/ltx_video.py:85
Method
save_model
(self, save_dir, state_dict)
models/cosmos_predict2.py:322
Method
save_model
(self, save_dir, diffusers_sd)
models/hunyuan_video.py:349
Method
save_model
(self, save_dir, diffusers_sd)
models/sd3.py:66
← previous
next →
1,001–1,100 of 1,157, ranked by callers