MCPcopy Create free account

hub / github.com/showlab/Show-o / functions

Functions766 in github.com/showlab/Show-o

⨍ Functions 766◇ Types & classes 141↳ Endpoints 3
Method__init__
( self, train_shards_path_or_url: Union[str, List[str]], tokenizer: PreTra
training/data.py:121
Method__init__
(self)
models/modeling_utils.py:136
Method__init__
Args: parameters (Iterable[torch.nn.Parameter]): The parameters to track. decay (float): The decay factor for the exp
models/training_utils.py:66
Method__init__
(self, *args, **kwargs)
models/logging.py:282
Method__init__
( self, w_clip_vit, vocab_size, llm_vocab_size, ll
models/modeling_showo.py:27
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None)
models/phi.py:80
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None, scaling_factor=1.0)
models/phi.py:119
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None, scaling_factor=1.0)
models/phi.py:139
Method__init__
(self, config: PhiConfig, layer_idx: Optional[int] = None)
models/phi.py:231
Method__init__
(self, *args, **kwargs)
models/phi.py:408
Method__init__
(self, *args, **kwargs)
models/phi.py:618
Method__init__
(self, config: PhiConfig, layer_idx: int)
models/phi.py:740
Method__init__
(self, config: PhiConfig)
models/phi.py:927
Method__init__
(self, config)
models/phi.py:1086
Method__init__
(self, config)
models/phi.py:1295
Method__init__
(self, config: PhiConfig)
models/phi.py:1411
Method__init__
( self, in_channels, )
models/common_modules.py:44
Method__init__
(self, in_channels, with_conv)
models/common_modules.py:74
Method__init__
( self, dim, dim_out=None, kernel_size=3, )
models/common_modules.py:107
Method__init__
(self, dim, dim_out=None)
models/common_modules.py:134
Method__init__
(self, in_channels)
models/common_modules.py:169
Method__init__
(self, fn: nn.Module)
models/common_modules.py:227
Method__init__
( self, chan_in, chan_out, kernel_size: Union[int, Tuple[int, int, int]],
models/common_modules.py:240
Method__init__
( self, *, in_channels, out_channels=None, conv_shortcut=False,
models/common_modules.py:299
Method__init__
(self, vision_tower, args, delay_load=False)
models/clip_encoder.py:86
Method__init__
(self, ch: int = 128, ch_mult: List[int] = [1, 2, 2, 4, 4],
models/modeling_magvitv2.py:62
Method__init__
(self, num_codebook_entry: int = -1, codebook_dim: int = 13, beta: float = 0
models/modeling_magvitv2.py:173
Method__init__
(self, ch: int = 128, ch_mult: List[int] = [1, 1, 2, 2, 4], num_res_blocks:
models/modeling_magvitv2.py:278
Method__iter__
(self)
parquet/refinedweb_dataset.py:38
Method__iter__
(self)
show-o2/datasets/mixed_dataloader.py:50
Method__iter__
(self)
show-o2/models/my_logging.py:285
Method__iter__
(self)
models/logging.py:285
Method__len__
(self)
llava/llava_data_vq_unified.py:152
Method__len__
(self)
llava/llava_pretrain_data.py:78
Method__len__
(self)
llava/llava_instruct_data.py:152
Method__len__
(self)
show-o2/datasets/image_text_dataset.py:95
Method__len__
(self)
show-o2/datasets/vist_dataset.py:170
Method__next__
(self)
show-o2/datasets/mixed_dataloader.py:131
Method_basic_init
(module)
show-o2/models/modeling_showo2_qwen2_5.py:125
Function_count_params
(module)
show-o2/utils.py:58
Function_f
(dictionary)
training/data.py:56
Method_find_mismatched_keys
( state_dict, model_state_dict, loaded_keys, ignore_mismatched
show-o2/models/modeling_utils.py:887
Method_find_mismatched_keys
( state_dict, model_state_dict, loaded_keys, ignore_mismatched
models/modeling_utils.py:889
Method_fn
(t, x)
show-o2/transport/integrators.py:110
Method_get_no_split_modules
Get the modules of the model that should not be spit when using device_map. We iterate through the modules to get the underlying `_no
show-o2/models/modeling_utils.py:978
Method_get_no_split_modules
Get the modules of the model that should not be spit when using device_map. We iterate through the modules to get the underlying `_no
models/modeling_utils.py:980
Method_get_signature_keys
(cls, obj)
show-o2/models/modeling_utils.py:969
Method_get_signature_keys
(cls, obj)
models/modeling_utils.py:971
Method_init_weights
(self, module)
show-o2/models/qwen2.py:697
Method_init_weights
Initialize the weights
show-o2/models/modeling_siglip.py:700
Method_init_weights
(self, module)
models/phi.py:833
Method_likelihood_drift
(x, t, model, **model_kwargs)
show-o2/transport/transport.py:465
Method_reorder_cache
(past_key_values, beam_idx)
models/phi.py:1269
Function_reset_library_root_logger
()
show-o2/models/my_logging.py:90
Function_reset_library_root_logger
()
models/logging.py:90
Method_sample
(init, model, **model_kwargs)
show-o2/transport/transport.py:368
Method_sample_fn
(x, model, **model_kwargs)
show-o2/transport/transport.py:495
Method_set_cos_sin_cache
(self, seq_len, device, dtype)
models/phi.py:123
Method_set_cos_sin_cache
(self, seq_len, device, dtype)
models/phi.py:143
Method_set_gradient_checkpointing
(self, module, value=False)
show-o2/models/modeling_showo2_qwen2_5.py:110
Method_set_gradient_checkpointing
(self, module, value=False)
show-o2/models/modeling_semantic_layers.py:62
Method_set_gradient_checkpointing
(self, module, value=False)
models/modeling_showo.py:56
Method_update_causal_mask
( self, attention_mask: torch.Tensor, input_tensor: torch.Tensor, cache_positi
show-o2/models/qwen2.py:946
Functionadd_handler
adds a handler to the HuggingFace muse' root logger.
show-o2/models/my_logging.py:201
Functionadd_handler
adds a handler to the HuggingFace muse' root logger.
models/logging.py:201
Methodbody_fn
(x, t, model, **model_kwargs)
show-o2/transport/transport.py:226
Functionbroadcast
(tensor, src=0)
models/misc.py:37
Functionbuild_logger
(logger_name, logger_filename)
llava/llava/utils.py:17
Functioncausal_attn_mask_naive
(B, LEN, device, inverted=True)
show-o2/models/omni_attention.py:84
Methodcausal_mask
(self, b, h, q_idx, kv_idx)
training/omni_attention.py:48
Functioncollate_fn
( instances, tokenizer=None, max_length=77, )
llava/llava_data_vq_unified.py:192
Functioncollate_fn
( instances, tokenizer=None, max_length=77, )
llava/llava_pretrain_data.py:112
Functioncollate_fn
( instances, tokenizer=None, max_length=77, )
llava/llava_instruct_data.py:192
Methodcollate_fn
(self, batch)
parquet/refinedweb_dataset.py:55
Methodcollate_fn
(self, batch)
show-o2/datasets/mmu_dataset.py:225
Methodcollate_fn
(self, batch)
show-o2/datasets/image_text_dataset.py:175
Methodcollate_fn
Collate function to batch data.
show-o2/datasets/vist_dataset.py:225
Methodcollate_fn
(self, batch)
show-o2/datasets/imagenet_dataset.py:179
Methodcollate_fn
(self, batch)
training/imagenet_dataset.py:69
Methodcompute_alpha_t
Compute coefficient of x1
show-o2/transport/path.py:160
Methodcompute_alpha_t
Compute coefficient of x1
show-o2/transport/path.py:189
Methodcompute_d_alpha_alpha_ratio_t
Special purposed function for computing numerical stabled d_alpha_t / alpha_t
show-o2/transport/path.py:174
Methodcompute_d_alpha_alpha_ratio_t
Special purposed function for computing numerical stabled d_alpha_t / alpha_t
show-o2/transport/path.py:201
Methodcompute_drift
Compute the drift term of the SDE
show-o2/transport/path.py:178
Methodcompute_sigma_t
Compute coefficient of x0
show-o2/transport/path.py:167
Methodcompute_sigma_t
Compute coefficient of x0
show-o2/transport/path.py:195
Methodconfig
(self)
models/clip_encoder.py:66
Functioncosine_schedule
(t)
models/sampling.py:39
Functioncreate_attention_mask_lvg
(sequence, pad_id=128256, soi_id=128257, eoi_id=128258, return_inverse_mask=True)
training/prompting_utils.py:513
Functioncreate_attention_mask_lvg_v2
(sequence, pad_id=128256, soi_id=128257, eoi_id=128258, sot_id=1000, eot_id=1001, return_inverse_mask=True)
training/prompting_utils.py:541
Functioncross_entropy_per_percent_masked_bucket
(logits, labels, input_ids, mask_id, output_size, label_smoothing)
models/training_utils.py:346
Methoddevice
(self)
models/clip_encoder.py:62
Methoddict
(self)
llava/llava/conversation.py:192
Methoddiffusion_fn
(x, t)
show-o2/transport/transport.py:277
Functiondisable_default_handler
Disable the default handler of the HuggingFace muse' root logger.
show-o2/models/my_logging.py:183
Functiondisable_default_handler
Disable the default handler of the HuggingFace muse' root logger.
models/logging.py:183
Methoddisable_gradient_checkpointing
Deactivates gradient checkpointing for the current model (may be referred to as *activation checkpointing* or *checkpoint activations
show-o2/models/modeling_utils.py:173
Methoddisable_gradient_checkpointing
Deactivates gradient checkpointing for the current model (may be referred to as *activation checkpointing* or *checkpoint activations
models/modeling_utils.py:173
Methoddisable_npu_flash_attention
r""" disable npu flash attention from torch_npu
show-o2/models/modeling_utils.py:204
Methoddisable_npu_flash_attention
r""" disable npu flash attention from torch_npu
models/modeling_utils.py:204
← previousnext →401–500 of 766, ranked by callers