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
get_idxs
(self)
optimizers/projectors/uniform_projector.py:39
Method
get_loss_fn
(self)
models/base.py:707
Method
get_loss_fn
(self)
models/cosmos.py:278
Method
get_loss_fn
(self)
models/sdxl.py:630
Method
get_loss_fn
(self)
models/ltx_video.py:203
Method
get_loss_fn
(self)
models/cosmos_predict2.py:491
Method
get_loss_fn
(self)
models/ltx2.py:376
Function
get_optimizer
(model_parameters)
train.py:640
Method
get_param_groups
(self, parameters)
models/base.py:703
Method
get_param_groups
(self, parameters)
models/sdxl.py:602
Method
get_param_groups
(self, parameters)
models/cosmos_predict2.py:448
Method
get_preprocess_media_file_fn
(self)
models/hunyuan_video_15.py:28
Method
get_preprocess_media_file_fn
(self)
models/base.py:642
Method
get_preprocess_media_file_fn
(self)
models/cosmos.py:199
Method
get_preprocess_media_file_fn
(self)
models/ltx_video.py:88
Method
get_preprocess_media_file_fn
(self)
models/cosmos_predict2.py:326
Method
get_preprocess_media_file_fn
(self)
models/hunyuan_video.py:352
Method
get_preprocess_media_file_fn
(self)
models/ltx2.py:278
Method
get_preprocess_media_file_fn
(self)
models/krea2.py:73
Method
get_preprocess_media_file_fn
(self)
models/qwen_image.py:293
Method
get_preprocess_media_file_fn
(self)
models/wan/wan.py:267
Method
get_text_encoders
(self)
models/hidream.py:129
Method
get_text_encoders
(self)
models/chroma.py:173
Method
get_text_encoders
(self)
models/auraflow.py:81
Method
get_text_encoders
(self)
models/lumina_2.py:92
Method
get_text_encoders
(self)
models/base.py:586
Method
get_text_encoders
(self)
models/cosmos.py:190
Method
get_text_encoders
(self)
models/sdxl.py:426
Method
get_text_encoders
(self)
models/ltx_video.py:76
Method
get_text_encoders
(self)
models/cosmos_predict2.py:310
Method
get_text_encoders
(self)
models/hunyuan_video.py:340
Method
get_text_encoders
(self)
models/hunyuan_image.py:122
Method
get_text_encoders
(self)
models/omnigen2.py:64
Method
get_text_encoders
(self)
models/flux.py:219
Method
get_text_encoders
(self)
models/qwen_image.py:281
Method
get_text_encoders
(self)
models/wan/wan.py:251
Method
get_vae
(self)
models/hidream.py:126
Method
get_vae
(self)
models/chroma.py:170
Method
get_vae
(self)
models/auraflow.py:78
Method
get_vae
(self)
models/lumina_2.py:89
Method
get_vae
(self)
models/base.py:583
Method
get_vae
(self)
models/cosmos.py:187
Method
get_vae
(self)
models/sdxl.py:423
Method
get_vae
(self)
models/ltx_video.py:73
Method
get_vae
(self)
models/cosmos_predict2.py:307
Method
get_vae
(self)
models/hunyuan_video.py:337
Method
get_vae
(self)
models/hunyuan_image.py:119
Method
get_vae
(self)
models/omnigen2.py:61
Method
get_vae
(self)
models/flux.py:216
Method
get_vae
(self)
models/qwen_image.py:278
Method
get_vae
(self)
models/wan/wan.py:246
Function
guidance_embed_bypass_forward
(self, timestep, guidance, pooled_projection)
models/flux.py:130
Function
init
(queue)
utils/dataset.py:115
Function
init_reductions
()
utils/reduction.py:629
Method
init_state
(self, group, p, gindex, pindex)
optimizers/adamw_8bit.py:12
Method
init_weight
(self, conv)
models/wan/vae2_2.py:171
Method
init_weight
(self, conv)
models/wan/vae2_1.py:158
Method
init_weight2
(self, conv)
models/wan/vae2_2.py:182
Method
init_weight2
(self, conv)
models/wan/vae2_1.py:170
Function
init_weights
(m)
models/wan/t5.py:21
Method
init_weights
(self)
models/llm_adapter.py:159
Function
latents_map_fn
(example, rank)
utils/dataset.py:1062
Method
load_adapter_weights
(self, adapter_path)
models/base.py:615
Method
load_adapter_weights
(self, adapter_path)
models/sdxl.py:476
Method
load_adapter_weights
(self, adapter_path)
models/hunyuan_image.py:139
Method
load_and_fuse_adapter
(self, path)
models/base.py:636
Method
load_diffusion_model
(self)
models/hidream.py:71
Method
load_diffusion_model
(self)
models/chroma.py:148
Method
load_diffusion_model
(self)
models/auraflow.py:58
Method
load_diffusion_model
(self)
models/base.py:556
Method
load_diffusion_model
(self)
models/cosmos.py:166
Method
load_diffusion_model
(self)
models/ltx_video.py:58
Method
load_diffusion_model
(self)
models/cosmos_predict2.py:259
Method
load_diffusion_model
(self)
models/hunyuan_video.py:302
Method
load_diffusion_model
(self)
models/ltx2.py:271
Method
load_diffusion_model
(self)
models/hunyuan_image.py:90
Method
load_diffusion_model
(self)
models/omnigen2.py:49
Method
load_diffusion_model
(self)
models/flux2.py:71
Method
load_diffusion_model
(self)
models/qwen_image.py:247
Method
load_diffusion_model
(self)
models/wan/wan.py:214
Method
load_fn
()
models/base.py:485
Method
load_fn
()
models/ltx2.py:204
Function
load_official_video_vae
()
tools/cosmos_vae_test.py:71
Method
load_state_dict
(self, state_dict)
utils/dataset.py:1408
Method
load_state_dict
(self, sd)
optimizers/generic_optim.py:642
Method
load_state_dict
(self, state_dict, strict=True)
optimizers/automagic.py:376
Method
load_state_dict
(self, state_dict)
optimizers/gradient_release.py:19
Function
load_text_encoder
( text_encoder_type, text_encoder_precision=None, text_encoder_path=None, logger=None, dev
utils/patches.py:72
Function
log_duration
(name)
utils/common.py:60
Method
loss_fn
(output, label)
models/base.py:365
Method
loss_fn
(output, label)
models/base.py:708
Method
loss_fn
(output, label)
models/cosmos.py:279
Method
loss_fn
(output, label)
models/sdxl.py:631
Method
loss_fn
(output, label)
models/ltx_video.py:204
Method
loss_fn
(output, label)
models/cosmos_predict2.py:492
Method
loss_fn
(output, label)
models/ltx2.py:377
Method
model_specific_dataset_config_validation
(self, dataset_config)
models/base.py:698
Method
model_specific_dataset_config_validation
(self, dataset_config)
models/cosmos.py:175
Method
move_blocks
(bidx_to_cpu, block_to_cpu, bidx_to_cuda, block_to_cuda)
utils/offloading.py:147
Method
new_load_fn
()
models/flux2.py:60
← previous
next →
901–1,000 of 1,157, ranked by callers