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

Method__getattr__

(self, name)

models/hidream.py:360

Method__getattr__

(self, name)

models/hidream.py:483

models/ernie_image.py:130

Method__getattr__

(self, name)

models/ernie_image.py:216

Method__getattr__

(self, name)

models/hunyuan_video_15.py:224

Method__getattr__

(self, name)

models/hunyuan_video_15.py:312

Method__getattr__

(self, name)

models/z_image.py:134

Method__getattr__

(self, name)

models/z_image.py:246

Method__getattr__

(self, name)

models/auraflow.py:55

Method__getattr__

(self, name)

models/auraflow.py:271

Method__getattr__

(self, name)

models/lumina_2.py:202

Method__getattr__

(self, name)

models/lumina_2.py:331

models/ltx_video.py:55

Method__getattr__

(self, name)

models/ltx_video.py:230

Method__getattr__

(self, name)

models/cosmos_predict2.py:622

Method__getattr__

(self, name)

models/hunyuan_video.py:334

models/hunyuan_image.py:395

Method__getattr__

(self, name)

models/ideogram4.py:145

Method__getattr__

(self, name)

models/ideogram4.py:249

models/omnigen2.py:46

Method__getattr__

(self, name)

models/omnigen2.py:307

models/qwen_image.py:275

Method__getattr__

(self, name)

models/qwen_image.py:589

Method__getattr__

(self, name)

models/wan/wan.py:243

Method__getattr__

(self, name)

models/wan/wan.py:537

utils/reduction.py:55

(self, z_channels: int, z_factor: int, embedding_dim: int, **kwargs)

tools/cosmos_vae_test.py:33

Method__init__

(self, args, config, is_adapter, save_root, model, train_dataloader, model_engine, pipeline_model)

utils/saver.py:48

Method__init__

(self, storage)

utils/reduction.py:36

Method__init__

(self)

utils/reduction.py:67

Method__init__

( self, block_type: str, blocks: list[nn.Module], num_blocks: int, blo

utils/offloading.py:189

Method__init__

(self, path: str, fingerprint: str, shard_size_gb=1)

utils/cache.py:11

Method__init__

(self, te_dataset, flattened_captions)

utils/dataset.py:164

Method__init__

(self, metadata_dataset, directory_config, size_bucket, cache_base, directory_dataset)

(self, ar_frames, resolutions, metadata_dataset, directory_config, cache_base, round_to_multiple, directory_da

utils/dataset.py:399

Method__init__

(self, directory_config, dataset_config, model_name, framerate=None, round_to_multiple=32, skip_dataset_valida

utils/dataset.py:448

Method__init__

(self, model, regenerate_cache=False, trust_cache=False, caching_batch_size=1, keep_models_loaded=False)

utils/dataset.py:1133

Method__init__

(self, dataset, model_engine, gradient_accumulation_steps, model, num_dataloader_workers=1)

utils/dataset.py:1288

Method__init__

(self, n, dataset_length)

utils/dataset.py:1422

Method__init__

(self, *args, manual_partition_split=None, **kwargs)

utils/pipeline.py:12

Method__init__

(self, *args, stabilize=False, **kwargs)

optimizers/adamw_8bit.py:7

Method__init__

(self, data: Tensor, *args, **kwargs)

optimizers/optimizer_utils.py:223

Method__init__

( self, params: Iterable, lr: float = 1e-3, # set beta2 = 1 to

optimizers/generic_optim.py:298

Method__init__

( self, params, lr=1e-6, # lr is start lr min_lr=1e-7, max_lr=1e-3,

optimizers/automagic.py:35

Method__init__

(self, optimizers)

optimizers/gradient_release.py:6

Method__init__

(self, rank, verbose=False, update_proj_gap=200, scale=1.0, proj_type='std', param_shape=None)

optimizers/projectors/topk_norm_projector.py:20

Method__init__

(self, rank, verbose=False, update_proj_gap=200, scale=1.0, proj_type='std', approx_svd=False,

optimizers/projectors/svd_projector.py:56

Method__init__

(self, rank, verbose=False, update_proj_gap=200, scale=1.0, param_shape=None)

optimizers/projectors/uniform_projector.py:8

models/hidream.py:351

Method__init__

(self, block, block_idx, global_block_idx, offloader)

models/hidream.py:415

Method__init__

(self, block, block_idx, global_block_idx, offloader)

models/hidream.py:450

Method__init__

(self, model)

models/hidream.py:478

(self, block, idx, offloader)

models/chroma.py:411

Method__init__

(self, block, idx, offloader)

models/chroma.py:462

Method__init__

(self, *args, **kwargs)

models/ernie_image.py:21

Method__init__

(self, model)

models/ernie_image.py:120

Method__init__

(self, layer, block_idx, offloader)

models/ernie_image.py:192

Method__init__

(self, *args, **kwargs)

models/hunyuan_video_15.py:24

Method__init__

(self, model)

models/hunyuan_video_15.py:210

Method__init__

(self, layer, block_idx, offloader)

models/hunyuan_video_15.py:289

Method__init__

(self, *args, **kwargs)

models/z_image.py:25

Method__init__

(self, model)

models/z_image.py:124

Method__init__

(self, layer, block_idx, offloader)

models/z_image.py:223

Method__init__

(self, config)

models/auraflow.py:24

Method__init__

(self, model)

models/auraflow.py:188

Method__init__

(self, block)

models/auraflow.py:222

Method__init__

(self, block, block_idx, offloader)

models/auraflow.py:247

Method__init__

(self, config)

models/lumina_2.py:43

Method__init__

(self, model)

models/lumina_2.py:192

Method__init__

(self, block)

models/lumina_2.py:311

Method__init__

(self, dim: int, eps: float = 1e-5)

models/cosmos_predict2_modeling.py:227

Method__init__

(self, d_model: int, d_ff: int)

models/cosmos_predict2_modeling.py:246

Method__init__

( self, query_dim: int, context_dim: Optional[int] = None, n_heads: int = 8,

models/cosmos_predict2_modeling.py:341

Method__init__

(self)

models/cosmos_predict2_modeling.py:464

← previous next →401–500 of 1,157, ranked by callers