MCPcopy Create free account

hub / github.com/JiangkaiWu/Promptus / functions

Functions645 in github.com/JiangkaiWu/Promptus

⨍ Functions 645◇ Types & classes 174↳ Endpoints 3
Method__init__
(self, watermark)
sgm/inference/helpers.py:17
Method__init__
(self, discretization, strength: float = 1.0)
sgm/inference/helpers.py:84
Method__init__
( self, network_config, denoiser_config, first_stage_config, condition
sgm/models/diffusion.py:20
Method__init__
( self, ema_decay: Union[None, float] = None, monitor: Union[None, str] = None,
sgm/models/autoencoder.py:29
Method__init__
( self, *args, encoder_config: Dict, decoder_config: Dict, loss_config
sgm/models/autoencoder.py:109
Method__init__
(self, embed_dim: int, **kwargs)
sgm/models/autoencoder.py:438
Method__init__
( self, embed_dim: int, n_embed: int, sane_index_shape: bool = False,
sgm/models/autoencoder.py:524
Method__init__
(self, *args, **kwargs)
sgm/models/autoencoder.py:550
Method__init__
( self, model: nn.Module, shape: Union[None, Tuple[int, int], List[int]] = (16, 16),
sgm/models/autoencoder.py:568
Method__init__
(self, **kwargs)
sgm/models/autoencoder.py:607
Method__len__
(self)
sgm/data/cifar10.py:16
Method__len__
(self)
sgm/data/mnist.py:16
Method__str__
(self)
quantization.py:92
Method_forward
(self, x, context=None)
sgm/modules/attention.py:613
Method_forward
(self, x: th.Tensor)
sgm/modules/diffusionmodules/openaimodel.py:396
Method_load_from_state_dict
(self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys,
quantization.py:84
Method_make_attn
(self)
sgm/modules/autoencoding/temporal_ae.py:324
Method_make_conv
(self)
sgm/modules/autoencoding/temporal_ae.py:334
Method_make_resblock
(self)
sgm/modules/autoencoding/temporal_ae.py:340
Methodadd_to_map
(refit_dict, name, values)
tensorrt_acceleration/utilities.py:108
Methodbackward
(ctx, *output_grads)
sgm/modules/diffusionmodules/util.py:120
Functionbenchmark_attn
()
scripts/tests/attention.py:10
Methodconfigure_optimizers
(self)
sgm/models/diffusion.py:217
Methodconfigure_optimizers
(self)
sgm/models/autoencoder.py:98
Methodconfigure_optimizers
(self)
sgm/models/autoencoder.py:363
Methodconvert_int64
(arr)
tensorrt_acceleration/utilities.py:102
Methoddecode
(self, *args, **kwargs)
sgm/models/autoencoder.py:89
Methoddecode
(self, z: torch.Tensor, **decoder_kwargs)
sgm/models/autoencoder.py:490
Methoddecode
(self, x: Any, *args, **kwargs)
sgm/models/autoencoder.py:562
Functiondecrement_counter
()
scripts/demo/turbo.py:190
Functiondenoiser
(input, sigma, c)
inversion.py:87
Functiondenoiser
(input, sigma, c)
generation.py:97
Functiondenoiser
(input, sigma, c)
scripts/sampling/simple_video_sample.py:172
Functiondenoiser
(input, sigma, c)
scripts/demo/turbo.py:132
Functiondenoiser
(input, sigma, c)
scripts/demo/gradio_app.py:194
Functiondenoiser
(input, sigma, c)
scripts/demo/streamlit_helpers.py:561
Functiondenoiser
(input, sigma, c)
scripts/demo/turbo_demo.py:140
Functiondisabled_train
Overwrite model.train with this function to make sure train/eval mode does not change anymore.
sgm/util.py:14
Functiondo_autocast
(*args, **kwargs)
sgm/util.py:53
Methodencode
(self, text)
sgm/modules/encoders/modules.py:285
Methodencode
(self, text)
sgm/modules/encoders/modules.py:327
Methodencode
(self, text)
sgm/modules/encoders/modules.py:392
Methodencode
(self, text)
sgm/modules/encoders/modules.py:494
Methodencode
(self, text)
sgm/modules/encoders/modules.py:575
Methodencode
(self, text)
sgm/modules/encoders/modules.py:736
Methodencode
(self, text)
sgm/modules/encoders/modules.py:759
Methodencode
(self, x)
sgm/modules/encoders/modules.py:824
Methodencode
(self, *args, **kwargs)
sgm/models/autoencoder.py:85
Methodencode
( self, x: torch.Tensor, return_reg_log: bool = False )
sgm/models/autoencoder.py:468
Methodencode
(self, x: Any, *args, **kwargs)
sgm/models/autoencoder.py:558
Methodencode_with_transformer_token_embedding
(self, x)
sgm/modules/encoders/modules.py:554
Functionfn
(tau)
sgm/modules/diffusionmodules/sampling_utils.py:11
Methodforward
(ctx, x, qparam)
quantization.py:9
Methodforward
(self, img)
lossbuilder.py:23
Methodforward
(self, x)
lossbuilder.py:31
Methodforward
(self, x, context=None, timesteps=None)
sgm/modules/video_attention.py:8
Methodforward
( self, x: torch.Tensor, context: torch.Tensor = None, timesteps: int = None )
sgm/modules/video_attention.py:101
Methodforward
( self, x: torch.Tensor, context: Optional[torch.Tensor] = None, time_context:
sgm/modules/video_attention.py:230
Methodforward
(self, model)
sgm/modules/ema.py:33
Methodforward
(self, x)
sgm/modules/attention.py:92
Methodforward
(self, x)
sgm/modules/attention.py:112
Methodforward
(self, x)
sgm/modules/attention.py:139
Methodforward
(self, x: torch.Tensor)
sgm/modules/attention.py:179
Methodforward
(self, x)
sgm/modules/attention.py:229
Methodforward
( self, x, context=None, mask=None, additional_tokens=None, n_
sgm/modules/attention.py:281
Methodforward
( self, x, context=None, mask=None, additional_tokens=None, n_
sgm/modules/attention.py:373
Methodforward
( self, x, context=None, additional_tokens=None, n_times_crossframe_attn_in_self=0 )
sgm/modules/attention.py:527
Methodforward
(self, x, context=None)
sgm/modules/attention.py:608
Methodforward
(self, x, context=None)
sgm/modules/attention.py:702
Methodforward
( self, x: torch.Tensor, context: Optional[torch.Tensor] = None, )
sgm/modules/attention.py:752
Methodforward
( self, network: nn.Module, denoiser: Denoiser, conditioner: GeneralConditione
sgm/modules/diffusionmodules/loss.py:48
Methodforward
(self, x: th.Tensor)
sgm/modules/diffusionmodules/openaimodel.py:43
Methodforward
Apply the module to `x` given `emb` timestep embeddings.
sgm/modules/diffusionmodules/openaimodel.py:60
Methodforward
( self, x: th.Tensor, emb: th.Tensor, context: Optional[th.Tensor] = None,
sgm/modules/diffusionmodules/openaimodel.py:72
Methodforward
(self, x: th.Tensor)
sgm/modules/diffusionmodules/openaimodel.py:139
Methodforward
(self, x: th.Tensor)
sgm/modules/diffusionmodules/openaimodel.py:204
Methodforward
Apply the block to a Tensor, conditioned on a timestep embedding. :param x: an [N x C x ...] Tensor of features. :param emb:
sgm/modules/diffusionmodules/openaimodel.py:316
Methodforward
(self, x: th.Tensor, **kwargs)
sgm/modules/diffusionmodules/openaimodel.py:393
Methodforward
Apply QKV attention. :param qkv: an [N x (H * 3 * C) x T] tensor of Qs, Ks, and Vs. :return: an [N x (H * C) x T] tensor afte
sgm/modules/diffusionmodules/openaimodel.py:414
Methodforward
Apply QKV attention. :param qkv: an [N x (3 * H * C) x T] tensor of Qs, Ks, and Vs. :return: an [N x (H * C) x T] tensor afte
sgm/modules/diffusionmodules/openaimodel.py:442
Methodforward
(self, t: th.Tensor)
sgm/modules/diffusionmodules/openaimodel.py:468
Methodforward
Apply the model to an input batch. :param x: an [N x C x ...] Tensor of inputs. :param timesteps: a 1-D batch of timesteps.
sgm/modules/diffusionmodules/openaimodel.py:816
Methodforward
( ctx, run_function, length_tensors, length_non_tensors, tensor_keys,
sgm/modules/diffusionmodules/util.py:80
Methodforward
(ctx, run_function, length, *args)
sgm/modules/diffusionmodules/util.py:173
Methodforward
(self, x)
sgm/modules/diffusionmodules/util.py:275
Methodforward
( self, x_spatial: torch.Tensor, x_temporal: torch.Tensor, image_only_indicato
sgm/modules/diffusionmodules/util.py:358
Methodforward
( self, x: th.Tensor, emb: th.Tensor, num_video_frames: int, image_onl
sgm/modules/diffusionmodules/video_model.py:62
Methodforward
( self, network: nn.Module, input: torch.Tensor, sigma: torch.Tensor,
sgm/modules/diffusionmodules/denoiser.py:23
Methodforward
(self, *args, **kwargs)
sgm/modules/diffusionmodules/wrappers.py:19
Methodforward
( self, x: torch.Tensor, t: torch.Tensor, c: dict, **kwargs )
sgm/modules/diffusionmodules/wrappers.py:24
Methodforward
(self, x)
sgm/modules/diffusionmodules/model.py:67
Methodforward
(self, x)
sgm/modules/diffusionmodules/model.py:84
Methodforward
(self, x, temb)
sgm/modules/diffusionmodules/model.py:131
Methodforward
(self, x, **kwargs)
sgm/modules/diffusionmodules/model.py:197
Methodforward
(self, x, **kwargs)
sgm/modules/diffusionmodules/model.py:261
Methodforward
(self, x, context=None, mask=None, **unused_kwargs)
sgm/modules/diffusionmodules/model.py:269
Methodforward
(self, x)
sgm/modules/diffusionmodules/model.py:576
Methodforward
(self, z, **kwargs)
sgm/modules/diffusionmodules/model.py:715
Methodforward
(self, x, temb, skip_video=False, timesteps=None)
sgm/modules/autoencoding/temporal_ae.py:64
Methodforward
(self, x, timesteps, skip_video=False)
sgm/modules/autoencoding/temporal_ae.py:142
← previousnext →401–500 of 645, ranked by callers