Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Little-Podi/AdaWorld
/ functions
Functions
471 in github.com/Little-Podi/AdaWorld
⨍
Functions
471
◇
Types & classes
120
↳
Endpoints
3
Method
__init__
( self, *args, encoder_config: Dict, decoder_config: Dict,
worldmodel/vwm/models/autoencoder.py:90
Method
__init__
(self, embed_dim: int, **kwargs)
worldmodel/vwm/models/autoencoder.py:141
Method
__init__
(self, **kwargs)
worldmodel/vwm/models/autoencoder.py:212
Method
__len__
(self)
lam/lam/dataset.py:149
Method
__len__
(self)
lam/lam/dataset.py:282
Method
__len__
(self)
lam/lam/dataset.py:425
Method
__len__
(self)
worldmodel/vwm/data/dataset.py:41
Method
__len__
(self)
worldmodel/vwm/data/dataset.py:164
Method
__len__
(self)
worldmodel/vwm/data/dataset.py:278
Method
__len__
(self)
worldmodel/vwm/data/dataset.py:415
Method
_make_conv
(self)
worldmodel/vwm/modules/autoencoding/temporal_ae.py:132
Method
_make_resblock
(self)
worldmodel/vwm/modules/autoencoding/temporal_ae.py:138
Function
apply_learned_rotations
(rotations, t, start_index=0, freq_ranges=None)
lam/lam/modules/embeddings.py:59
Function
apply_learned_rotations
(rotations, t, start_index=0, freq_ranges=None)
worldmodel/external/lam/modules/embeddings.py:59
Function
broadcat
(tensors, dim=-1)
lam/lam/modules/embeddings.py:21
Function
broadcat
(tensors, dim=-1)
worldmodel/external/lam/modules/embeddings.py:21
Method
configure_optimizers
(self)
lam/lam/model.py:166
Method
configure_optimizers
(self)
worldmodel/external/lam/model.py:147
Method
configure_optimizers
(self)
worldmodel/vwm/models/diffusion.py:178
Method
configure_optimizers
(self)
worldmodel/vwm/models/autoencoder.py:79
Method
decode
(self, *args, **kwargs)
worldmodel/vwm/models/autoencoder.py:72
Method
decode
(self, z: torch.Tensor, **decoder_kwargs)
worldmodel/vwm/models/autoencoder.py:182
Function
default_worker_init_fn
(worker_id: int)
lam/lam/dataset.py:25
Method
device
(self)
lam/lam/modules/embeddings.py:135
Function
disabled_train
Overwrite model.train with this function to make sure train/eval mode does not change anymore.
worldmodel/vwm/util.py:13
Function
divein
(*args, **kwargs)
worldmodel/train.py:749
Function
divein
(*args, **kwargs)
worldmodel/train_adapt.py:751
Function
do_autocast
(*args, **kwargs)
worldmodel/vwm/util.py:22
Method
ema_scope
(self, context=None)
worldmodel/vwm/models/autoencoder.py:53
Method
encode
(self, *args, **kwargs)
worldmodel/vwm/models/autoencoder.py:68
Method
encode
( self, x: torch.Tensor, return_reg_log: bool = False )
worldmodel/vwm/models/autoencoder.py:159
Method
extract_features
(self, x)
worldmodel/fvd_utils/pytorch_i3d.py:328
Method
extract_features_multiscale
(self, x)
worldmodel/fvd_utils/pytorch_i3d.py:340
Method
extract_pre_pool_features
(self, x)
worldmodel/fvd_utils/pytorch_i3d.py:334
Function
fn
(tau)
worldmodel/vwm/modules/diffusionmodules/sampling_utils.py:11
Method
forward
(self, x: Tensor)
lam/lam/modules/blocks.py:37
Method
forward
(self, x: Tensor, is_causal: bool = False)
lam/lam/modules/blocks.py:78
Method
forward
(self, x: Tensor)
lam/lam/modules/blocks.py:107
Method
forward
(self, x: Tensor, causal_temporal: bool = False)
lam/lam/modules/blocks.py:140
Method
forward
(self, x: Tensor)
lam/lam/modules/blocks.py:195
Method
forward
(self, x: Tensor)
lam/lam/modules/blocks.py:234
Method
forward
(self, x: Tensor, delta_psnr: bool = False)
lam/lam/modules/blocks.py:279
Method
forward
(self, x: Tensor)
lam/lam/modules/blocks.py:309
Method
forward
(self, batch: Dict)
lam/lam/modules/lam.py:92
Method
forward
(self, x)
worldmodel/fast_init_mlp.py:15
Method
forward
(self, x)
worldmodel/fvd_utils/pytorch_i3d.py:13
Method
forward
(self, x)
worldmodel/fvd_utils/pytorch_i3d.py:118
Method
forward
(self, x)
worldmodel/fvd_utils/pytorch_i3d.py:316
Method
forward
(self, x: Tensor)
worldmodel/external/lam/modules/blocks.py:35
Method
forward
(self, x: Tensor, is_causal: bool = False)
worldmodel/external/lam/modules/blocks.py:76
Method
forward
(self, x: Tensor)
worldmodel/external/lam/modules/blocks.py:105
Method
forward
(self, x: Tensor, causal_temporal: bool = False)
worldmodel/external/lam/modules/blocks.py:138
Method
forward
(self, x: Tensor)
worldmodel/external/lam/modules/blocks.py:193
Method
forward
(self, x: Tensor)
worldmodel/external/lam/modules/blocks.py:232
Method
forward
(self, batch: Dict)
worldmodel/external/lam/modules/lam.py:88
Method
forward
(self, x, context=None, timesteps=None)
worldmodel/vwm/modules/video_attention.py:8
Method
forward
(self, x: torch.Tensor, context: torch.Tensor = None, timesteps: int = None)
worldmodel/vwm/modules/video_attention.py:98
Method
forward
( self, x: torch.Tensor, context: Optional[torch.Tensor] = None,
worldmodel/vwm/modules/video_attention.py:226
Method
forward
(self, model)
worldmodel/vwm/modules/ema.py:33
Method
forward
(self, x)
worldmodel/vwm/modules/attention.py:75
Method
forward
(self, x)
worldmodel/vwm/modules/attention.py:111
Method
forward
( self, x, context=None, mask=None, additional_tok
worldmodel/vwm/modules/attention.py:162
Method
forward
( self, x, context=None, mask=None, additional_tok
worldmodel/vwm/modules/attention.py:233
Method
forward
(self, x, context=None, additional_tokens=None)
worldmodel/vwm/modules/attention.py:355
Method
forward
(self, x, context=None)
worldmodel/vwm/modules/attention.py:458
Method
forward
( self, network: nn.Module, denoiser: Denoiser, conditioner: G
worldmodel/vwm/modules/diffusionmodules/loss.py:40
Method
forward
Apply the module to 'x' given 'emb' timestep embeddings.
worldmodel/vwm/modules/diffusionmodules/openaimodel.py:20
Method
forward
( self, x: torch.Tensor, emb: torch.Tensor, context: Optional[
worldmodel/vwm/modules/diffusionmodules/openaimodel.py:31
Method
forward
(self, x)
worldmodel/vwm/modules/diffusionmodules/openaimodel.py:79
Method
forward
(self, x)
worldmodel/vwm/modules/diffusionmodules/openaimodel.py:127
Method
forward
Apply the block to a Tensor, conditioned on a timestep embedding. :param x: An [N x C x ...] Tensor of features. :param emb:
worldmodel/vwm/modules/diffusionmodules/openaimodel.py:228
Method
forward
(self, t: torch.Tensor)
worldmodel/vwm/modules/diffusionmodules/openaimodel.py:276
Method
forward
(ctx, run_function, length, *args)
worldmodel/vwm/modules/diffusionmodules/util.py:37
Method
forward
(self, x)
worldmodel/vwm/modules/diffusionmodules/util.py:126
Method
forward
( self, x_spatial: torch.Tensor, x_temporal: torch.Tensor )
worldmodel/vwm/modules/diffusionmodules/util.py:202
Method
forward
( self, x: torch.Tensor, emb: torch.Tensor, num_frames: int
worldmodel/vwm/modules/diffusionmodules/video_model.py:59
Method
forward
( self, network: nn.Module, noised_input: torch.Tensor, sigma:
worldmodel/vwm/modules/diffusionmodules/denoiser.py:19
Method
forward
(self, *args, **kwargs)
worldmodel/vwm/modules/diffusionmodules/wrappers.py:20
Method
forward
( self, x: torch.Tensor, t: torch.Tensor, c: Dict,
worldmodel/vwm/modules/diffusionmodules/wrappers.py:25
Method
forward
(self, x)
worldmodel/vwm/modules/diffusionmodules/model.py:57
Method
forward
(self, x)
worldmodel/vwm/modules/diffusionmodules/model.py:72
Method
forward
(self, x, temb)
worldmodel/vwm/modules/diffusionmodules/model.py:111
Method
forward
(self, x, **kwargs)
worldmodel/vwm/modules/diffusionmodules/model.py:157
Method
forward
(self, x, **kwargs)
worldmodel/vwm/modules/diffusionmodules/model.py:208
Method
forward
(self, x, context=None, mask=None, **unused_kwargs)
worldmodel/vwm/modules/diffusionmodules/model.py:216
Method
forward
(self, z, **kwargs)
worldmodel/vwm/modules/diffusionmodules/model.py:459
Method
forward
(self, z: torch.Tensor)
worldmodel/vwm/modules/autoencoding/regularizer.py:14
Method
forward
(self, z: torch.Tensor)
worldmodel/vwm/modules/autoencoding/regularizer.py:30
Method
forward
(self, x, temb, skip_video=False, timesteps=None)
worldmodel/vwm/modules/autoencoding/temporal_ae.py:54
Method
forward
(self, input: torch.Tensor, **kwargs)
worldmodel/vwm/modules/autoencoding/temporal_ae.py:98
Method
forward
(self, batch: Dict, force_zero_embeddings: Optional[List] = None)
worldmodel/vwm/modules/encoders/modules.py:113
Method
forward
(self, image, no_dropout=False)
worldmodel/vwm/modules/encoders/modules.py:237
Method
forward
(self, x)
worldmodel/vwm/modules/encoders/modules.py:330
Method
forward
(self, x)
worldmodel/vwm/modules/encoders/modules.py:351
Method
forward
(self, vid)
worldmodel/vwm/modules/encoders/modules.py:362
Method
forward
(self, vid)
worldmodel/vwm/modules/encoders/modules.py:387
Method
forward
(self, vid)
worldmodel/vwm/modules/encoders/modules.py:405
Method
forward
(self, x)
worldmodel/vwm/modules/encoders/modules.py:421
Method
forward
(self, x, batch)
worldmodel/vwm/models/diffusion.py:116
Method
forward
( self, x: torch.Tensor, **additional_decode_kwargs )
worldmodel/vwm/models/autoencoder.py:132
← previous
next →
301–400 of 471, ranked by callers