MCPcopy Create free account

hub / github.com/Little-Podi/AdaWorld / functions

Functions471 in github.com/Little-Podi/AdaWorld

⨍ Functions 471◇ Types & classes 120↳ Endpoints 3
↓ 1 callersMethod_forward
( self, network: nn.Module, denoiser: Denoiser, cond: Dict,
worldmodel/vwm/modules/diffusionmodules/loss.py:51
↓ 1 callersMethod_forward
(self, x: torch.Tensor, emb: torch.Tensor)
worldmodel/vwm/modules/diffusionmodules/openaimodel.py:242
↓ 1 callersFunction_get_fp32_state_dict_from_zero2_checkpoint
(world_size, fp32_flat_groups, zero_model_states)
worldmodel/zero_to_fp32.py:319
↓ 1 callersFunction_get_fp32_state_dict_from_zero3_checkpoint
(world_size, fp32_flat_groups, zero_model_states)
worldmodel/zero_to_fp32.py:435
↓ 1 callersFunction_get_fp32_state_dict_from_zero_checkpoint
Returns fp32 state_dict reconstructed from ds checkpoint. Args: - ``ds_checkpoint_dir``: path to the deepspeed checkpoint folder (wh
worldmodel/zero_to_fp32.py:188
↓ 1 callersMethod_init_first_stage
(self, config)
worldmodel/vwm/models/diffusion.py:73
↓ 1 callersMethod_make_attn
(self)
worldmodel/vwm/modules/diffusionmodules/model.py:447
↓ 1 callersMethod_make_conv
(self)
worldmodel/vwm/modules/diffusionmodules/model.py:453
↓ 1 callersMethod_make_resblock
(self)
worldmodel/vwm/modules/diffusionmodules/model.py:450
↓ 1 callersFunction_zero2_merge_frozen_params
(state_dict, zero_model_states)
worldmodel/zero_to_fp32.py:214
↓ 1 callersFunction_zero2_merge_trainable_params
(state_dict, world_size, fp32_flat_groups, zero_model_states)
worldmodel/zero_to_fp32.py:246
↓ 1 callersFunction_zero3_merge_frozen_params
(state_dict, world_size, zero_model_states)
worldmodel/zero_to_fp32.py:345
↓ 1 callersFunction_zero3_merge_trainable_params
(state_dict, world_size, fp32_flat_groups, zero_model_states)
worldmodel/zero_to_fp32.py:381
↓ 1 callersFunctionappend_zero
(x)
worldmodel/vwm/util.py:82
↓ 1 callersFunctionatoi
(text)
worldmodel/zero_to_fp32.py:47
↓ 1 callersMethodattention
(self, h_: torch.Tensor)
worldmodel/vwm/modules/diffusionmodules/model.py:144
↓ 1 callersMethodattention
(self, h_: torch.Tensor)
worldmodel/vwm/modules/diffusionmodules/model.py:181
↓ 1 callersFunctionautocast
(f, enabled=True)
worldmodel/vwm/util.py:21
↓ 1 callersFunctionavg_pool_nd
Create a 1D, 2D, or 3D average pooling module.
worldmodel/vwm/modules/diffusionmodules/util.py:153
↓ 1 callersMethodbackward
(ctx, *output_grads)
worldmodel/vwm/modules/diffusionmodules/util.py:51
↓ 1 callersMethodbuild
(self)
worldmodel/fvd_utils/pytorch_i3d.py:312
↓ 1 callersMethodbuild_data_dict
(self, video: Tensor)
lam/lam/dataset.py:223
↓ 1 callersMethodbuild_data_dict
(self, video: Tensor)
lam/lam/dataset.py:357
↓ 1 callersMethodbuild_data_dict
(self, image_seq: List)
worldmodel/vwm/data/dataset.py:119
↓ 1 callersMethodbuild_data_dict
(self, image_seq: List, raw_action: torch.Tensor)
worldmodel/vwm/data/dataset.py:226
↓ 1 callersMethodbuild_data_dict
(self, image_seq: List, raw_action: List)
worldmodel/vwm/data/dataset.py:339
↓ 1 callersMethodcheck_frequency
(self, check_idx)
worldmodel/train.py:391
↓ 1 callersMethodcheck_frequency
(self, check_idx)
worldmodel/train_adapt.py:391
↓ 1 callersFunctionconvert_zero_checkpoint_to_fp32_state_dict
Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated ``state_dict`` file that can be loaded with ``torch.load(file)`` + ``load_stat
worldmodel/zero_to_fp32.py:503
↓ 1 callersFunctioncount_params
(model, verbose=False)
worldmodel/vwm/util.py:53
↓ 1 callersFunctiondataset2path
(dataset_name)
process_rtx.py:18
↓ 1 callersMethoddenoise
(self, x, denoiser, sigma, cond, uc)
worldmodel/vwm/modules/diffusionmodules/sampling.py:44
↓ 1 callersFunctiondo_sample
( model, sampler, value_dict, input_res: int = 256, force_uc_zero_embe
worldmodel/sample_utils.py:144
↓ 1 callersMethodencode
(self, videos: Tensor)
lam/lam/modules/lam.py:56
↓ 1 callersMethodencode
(self, videos: Tensor)
worldmodel/external/lam/modules/lam.py:54
↓ 1 callersMethodencode_with_vision_transformer
(self, img)
worldmodel/vwm/modules/encoders/modules.py:285
↓ 1 callersMethodeuler_step
(self, x, d, dt)
worldmodel/vwm/modules/diffusionmodules/sampling.py:68
↓ 1 callersMethodforward
(self, x)
worldmodel/fvd_utils/pytorch_i3d.py:76
↓ 1 callersMethodforward
( self, x: torch.Tensor, timesteps: torch.Tensor, context: Opt
worldmodel/vwm/modules/diffusionmodules/video_model.py:428
↓ 1 callersMethodforward
(self, x)
worldmodel/vwm/modules/diffusionmodules/model.py:332
↓ 1 callersMethodfreeze
(self)
worldmodel/vwm/modules/encoders/modules.py:231
↓ 1 callersMethodget_alpha
(self)
worldmodel/vwm/modules/diffusionmodules/util.py:190
↓ 1 callersMethodget_alpha
(self)
worldmodel/vwm/modules/autoencoding/temporal_ae.py:46
↓ 1 callersFunctionget_batch
(keys, value_dict)
worldmodel/sample_utils.py:199
↓ 1 callersFunctionget_discretization
(discretization)
worldmodel/sample_utils.py:111
↓ 1 callersFunctionget_guider
(guider, scale)
worldmodel/sample_utils.py:80
↓ 1 callersFunctionget_logits
(i3d, videos, device, batch_size=None)
worldmodel/fvd_utils/fvd_utils.py:23
↓ 1 callersMethodget_loss
(self, model_output, target, w)
worldmodel/vwm/modules/diffusionmodules/loss.py:70
↓ 1 callersFunctionget_model_state_files
(checkpoint_dir)
worldmodel/zero_to_fp32.py:89
↓ 1 callersMethodget_noised_input
( self, sigmas_bc: torch.Tensor, noise: torch.Tensor, input: t
worldmodel/vwm/modules/diffusionmodules/loss.py:28
↓ 1 callersFunctionget_optim_files
(checkpoint_dir)
worldmodel/zero_to_fp32.py:85
↓ 1 callersFunctionget_parser
(**parser_kwargs)
worldmodel/train.py:40
↓ 1 callersFunctionget_parser
(**parser_kwargs)
worldmodel/train_adapt.py:40
↓ 1 callersFunctionget_sample
(source_video_dict, target_video_dict)
worldmodel/sample.py:79
↓ 1 callersFunctionget_sampler
(sampler, steps, discretization_config, guider_config, n_context_frames)
worldmodel/sample_utils.py:121
↓ 1 callersMethodget_scale
(self, t: Tensor, seq_len: Union[int, None] = None, offset=0)
lam/lam/modules/embeddings.py:180
↓ 1 callersMethodget_scale
(self, t: Tensor, seq_len: Union[int, None] = None, offset=0)
worldmodel/external/lam/modules/embeddings.py:180
↓ 1 callersMethodget_sigma_gen
(self, num_sigmas)
worldmodel/vwm/modules/diffusionmodules/sampling.py:49
↓ 1 callersMethodget_sigmas
(self, n, device)
worldmodel/vwm/modules/diffusionmodules/discretizer.py:19
↓ 1 callersFunctioninit_model
(config, ckpt, load_ckpt: bool = True)
worldmodel/sample_utils.py:17
↓ 1 callersFunctioninit_sampling
( steps: int = 40, sampler: str = "EulerEDMSampler", discretization: str = "EDMShiftDi
worldmodel/sample_utils.py:97
↓ 1 callersMethodinstantiate_optimizer_from_config
(self, params, lr, cfg)
worldmodel/vwm/models/diffusion.py:175
↓ 1 callersMethodkl
(self, other=None)
worldmodel/vwm/modules/distributions/distributions.py:20
↓ 1 callersFunctionload_model_from_config
(config, ckpt: str = None)
worldmodel/sample_utils.py:23
↓ 1 callersMethodload_video_slice
( self, video_path: str, num_frames: int, start_frame: int = N
lam/lam/dataset.py:166
↓ 1 callersMethodload_video_slice
( self, video_path: str, num_frames: int, start_frame: int = N
lam/lam/dataset.py:299
↓ 1 callersMethodload_video_slice
( self, video_path: str, num_frames: int, start_frame: int = N
worldmodel/vwm/data/dataset.py:58
↓ 1 callersMethodload_video_slice
( self, video_path: str, num_frames: int )
worldmodel/vwm/data/dataset.py:180
↓ 1 callersMethodload_video_slice
( self, video_path: str, num_frames: int )
worldmodel/vwm/data/dataset.py:296
↓ 1 callersMethodlog_local
( self, save_dir, split, images, global_step,
worldmodel/train.py:321
↓ 1 callersMethodlog_local
( self, save_dir, split, images, global_step,
worldmodel/train_adapt.py:321
↓ 1 callersMethodmode
(self)
worldmodel/vwm/modules/distributions/distributions.py:45
↓ 1 callersFunctionparse_model_states
(files)
worldmodel/zero_to_fp32.py:93
↓ 1 callersFunctionparse_optim_states
(files, ds_checkpoint_dir)
worldmodel/zero_to_fp32.py:138
↓ 1 callersFunctionpatchify
(videos: Tensor, size: int)
lam/lam/modules/blocks.py:12
↓ 1 callersMethodpossibly_get_ucg_val
(self, embedder: AbstractEmbModel, batch: Dict)
worldmodel/vwm/modules/encoders/modules.py:104
↓ 1 callersMethodpossibly_quantize_c_noise
(self, c_noise: torch.Tensor)
worldmodel/vwm/modules/diffusionmodules/denoiser.py:16
↓ 1 callersMethodpossibly_quantize_sigma
(self, sigma: torch.Tensor)
worldmodel/vwm/modules/diffusionmodules/denoiser.py:13
↓ 1 callersMethodprepare_inputs
(self, x, s, c, uc)
worldmodel/vwm/modules/diffusionmodules/guiders.py:13
↓ 1 callersMethodprepare_sampling_loop
(self, x, cond, uc=None, num_steps=None)
worldmodel/vwm/modules/diffusionmodules/sampling.py:32
↓ 1 callersFunctionpreprocess
(videos, target_resolution)
worldmodel/fvd_utils/fvd_utils.py:11
↓ 1 callersMethodpreprocess
(self, x)
worldmodel/vwm/modules/encoders/modules.py:217
↓ 1 callersMethodreload_ckpt
(self, ckpt_path: str)
worldmodel/external/lam/model.py:55
↓ 1 callersFunctionrotate_half
(x)
lam/lam/modules/embeddings.py:26
↓ 1 callersFunctionrotate_half
(x)
worldmodel/external/lam/modules/embeddings.py:26
↓ 1 callersFunctionrun_fdm
(fdm_model, source_video_dict, target_video_dict)
worldmodel/sample.py:107
↓ 1 callersMethodsample
(self)
worldmodel/vwm/modules/distributions/distributions.py:16
↓ 1 callersMethodsampler_step
(self, sigma, next_sigma, denoiser, x, cond, uc=None, gamma=0.0)
worldmodel/vwm/modules/diffusionmodules/sampling.py:80
↓ 1 callersFunctionsave_checkpoint
(model, optimizer, epoch, loss, filename="mlp_init_weights.pth")
worldmodel/fast_init_mlp.py:21
↓ 1 callersFunctionsave_images_to_video
(images: list, output_file: str, fps: int = 10)
sample_retro.py:17
↓ 1 callersFunctionsave_images_to_video
(images: list, output_file: str, fps: int = 10)
sample_procgen.py:17
↓ 1 callersFunctionsave_images_to_video
(images: list, output_file: str, fps: int = 10)
process_rtx.py:26
↓ 1 callersFunctionsave_images_to_video
(images: list, output_file: str, fps: int = 10)
sample_stableretro.py:17
↓ 1 callersMethodscaled_dot_product_attention
( self, query: Tensor, key: Tensor, value: Tensor,
lam/lam/modules/blocks.py:60
↓ 1 callersMethodschedule
(self, n, **kwargs)
worldmodel/vwm/lr_scheduler.py:18
↓ 1 callersMethodschedule
(self, n, **kwargs)
worldmodel/vwm/lr_scheduler.py:62
↓ 1 callersMethodshared_step
(self, batch: Dict)
worldmodel/vwm/models/diffusion.py:122
↓ 1 callersFunctionto_d
(x, sigma, denoised)
worldmodel/vwm/modules/diffusionmodules/sampling_utils.py:34
↓ 1 callersFunctiontrace_sqrt_product
(sigma, sigma_v)
worldmodel/fvd_utils/fvd_utils.py:56
↓ 1 callersFunctiontrain_action_mlp
(model, data_loader, num_epochs=3000, learning_rate=0.01)
worldmodel/fast_init_mlp.py:32
← previousnext →101–200 of 471, ranked by callers