MCPcopy Create free account

hub / github.com/OpenDriveLab/ReSim / functions

Functions2,541 in github.com/OpenDriveLab/ReSim

⨍ Functions 2,541◇ Types & classes 556↳ Endpoints 3
↓ 1 callersFunctionattention
(query, key, value)
SwissArmyTransformer/tests/test_mea.py:20
↓ 1 callersMethodattention
(self, h_: torch.Tensor)
sat/sgm/modules/diffusionmodules/model.py:194
↓ 1 callersFunctionattention_fn_default
(query_layer, key_layer, value_layer, attention_mask, attention_dropout=None, log_atten
SwissArmyTransformer/sat/transformer_defaults.py:47
↓ 1 callersFunctionavg_pool_nd
Create a 1D, 2D, or 3D average pooling module.
sat/sgm/modules/diffusionmodules/util.py:268
↓ 1 callersFunctionbackward_step
Backward step.
SwissArmyTransformer/sat/training/deepspeed_training.py:533
↓ 1 callersFunctionbase
(pretrained=None, **kwargs)
SwissArmyTransformer/examples/yolos/models/backbone.py:445
↓ 1 callersMethodbatch_generate
(self, tokenizer, queries, max_length: int = 2048, num_beams=1, do_sample=True, top_p=0.7, temper
SwissArmyTransformer/examples/chatglm/chat_model.py:192
↓ 1 callersMethodbatch_generate
(self, tokenizer, queries, max_length: int = 8192, num_beams=1, do_sample=True, top_p=0.8, temper
SwissArmyTransformer/examples/chatglm2/chat_model.py:180
↓ 1 callersMethodbound
Bound `z`, an array of shape (..., d).
sat/sgm/modules/autoencoding/regularizers/finite_scalar_quantization.py:91
↓ 1 callersFunctionbox_cxcywh_to_xyxy
(x)
SwissArmyTransformer/examples/yolos/infer_util.py:79
↓ 1 callersFunctionbox_iou
(boxes1, boxes2)
SwissArmyTransformer/examples/yolos/util/box_ops.py:24
↓ 1 callersFunctionbox_xyxy_to_cxcywh
(x)
SwissArmyTransformer/examples/yolos/util/box_ops.py:16
↓ 1 callersMethodbpe
(self, token)
SwissArmyTransformer/sat/tokenization/glm/tokenization_gpt2.py:196
↓ 1 callersMethodbpe
(self, token)
SwissArmyTransformer/sat/tokenization/cogview/sp_tokenizer.py:46
↓ 1 callersFunctionbroadcat
(tensors, dim=-1)
sat/dit_video_concat.py:241
↓ 1 callersFunctionbuild
(args)
SwissArmyTransformer/examples/yolos/models/detector.py:277
↓ 1 callersMethodbuild_inputs
(self, tokenizer, query: str, history: List[Tuple[str, str]] = None)
SwissArmyTransformer/examples/chatglm2/chat_model.py:155
↓ 1 callersFunctionbuild_matcher
(args)
SwissArmyTransformer/examples/yolos/models/matcher.py:85
↓ 1 callersFunctionbuild_transform
(is_train, args)
SwissArmyTransformer/examples/mae/util/datasets.py:31
↓ 1 callersFunctionbuilder_closure
(member_name)
SwissArmyTransformer/sat/ops/ops_builder/__init__.py:28
↓ 1 callersFunctionbytes_to_unicode
Returns list of utf-8 byte and a corresponding list of unicode strings. The reversible bpe codes work on unicode strings. This means you
SwissArmyTransformer/sat/tokenization/glm/tokenization_gpt2.py:56
↓ 1 callersMethodcalculate_adaptive_weight
( self, nll_loss: torch.Tensor, g_loss: torch.Tensor, last_layer: torch.Tensor )
sat/sgm/modules/autoencoding/losses/discriminator_loss.py:188
↓ 1 callersFunctioncast_tuple
(t, length=1)
sat/sgm/modules/cp_enc_dec.py:28
↓ 1 callersFunctioncast_tuple
(t, length=1)
sat/sgm/modules/autoencoding/vqvae/movq_enc_3d.py:13
↓ 1 callersFunctioncast_tuple
(t, length=1)
sat/vae_modules/cp_enc_dec.py:26
↓ 1 callersFunctionchat
from https://huggingface.co/spaces/ysharma/Explore_llamav2_with_TGI/blob/main/app.py system_message = "\nYou are a helpful, respectful and honest
SwissArmyTransformer/examples/llama/split_model.py:10
↓ 1 callersFunctionchat
(model, tokenizer, max_length: int = 256, num_beams=1, top_p=0.7, top_k=0, temperature=0.95)
SwissArmyTransformer/examples/llama/chat_sat.py:8
↓ 1 callersFunctionchat
(model, tokenizer, max_length: int = 256, num_beams=1, top_p=0.7, top_k=0, temperature=0.95)
SwissArmyTransformer/examples/chatglm3/chat_sat.py:31
↓ 1 callersFunctionchat
(model, tokenizer, query: str, history: List[Tuple[str, str]] = None, max_length: int = 1024
SwissArmyTransformer/examples/chatglm/chat_sat.py:70
↓ 1 callersFunctionchat
(model, tokenizer, max_length: int = 256, top_p=0.7, top_k=0, temperature=0.95)
SwissArmyTransformer/examples/chatglm2/chat_sat.py:8
↓ 1 callersFunctionchat_history_to_prompt
(history, query)
SwissArmyTransformer/examples/mixtral-8x7b/chat_sat.py:60
↓ 1 callersMethodclean_up_tokenization
(out_string: str)
SwissArmyTransformer/sat/tokenization/glm/tokenization.py:310
↓ 1 callersMethodcluster_size_ema_update
(self, new_cluster_size)
sat/sgm/modules/autoencoding/regularizers/quantize.py:311
↓ 1 callersFunctioncode2img
Convert a batch of code to imgs Args: model: ... code: [b, h, w] or [b, h*w] LongTensor
SwissArmyTransformer/sat/tokenization/cogview/vqvae/api.py:225
↓ 1 callersMethodcodes_to_indices
Converts a `code` to an index in the codebook.
sat/sgm/modules/autoencoding/regularizers/finite_scalar_quantization.py:112
↓ 1 callersMethodcommand_exists
(self, cmd)
SwissArmyTransformer/sat/ops/ops_builder/builder.py:400
↓ 1 callersMethodcompute_bias
Compute binned relative position bias
SwissArmyTransformer/sat/model/official/t5_model.py:99
↓ 1 callersFunctioncompute_metrics
(eval_preds)
SwissArmyTransformer/examples/chatglm/finetune_chatglm.py:77
↓ 1 callersFunctionconfigure_logging
()
SwissArmyTransformer/sat/helpers.py:108
↓ 1 callersFunctionconv_gather_from_context_parallel_region
(input_, dim, kernel_size)
sat/sgm/modules/cp_enc_dec.py:287
↓ 1 callersFunctionconv_gather_from_context_parallel_region
(input_, dim, kernel_size)
sat/vae_modules/cp_enc_dec.py:348
↓ 1 callersFunctionconv_pass_from_last_rank
(input_, dim, kernel_size)
sat/sgm/modules/cp_enc_dec.py:291
↓ 1 callersFunctionconv_scatter_to_context_parallel_region
(input_, dim, kernel_size)
sat/sgm/modules/cp_enc_dec.py:283
↓ 1 callersFunctionconv_scatter_to_context_parallel_region
(input_, dim, kernel_size)
sat/vae_modules/cp_enc_dec.py:344
↓ 1 callersFunctionconvert_coco_poly_to_mask
(segmentations, height, width)
SwissArmyTransformer/examples/yolos/datasets_/voc.py:47
↓ 1 callersMethodconvert_id_to_token
(self, idx)
SwissArmyTransformer/sat/tokenization/cogview/sp_tokenizer.py:130
↓ 1 callersMethodconvert_ids_to_tokens
Converts a sequence of ids in BPE tokens using the vocab.
SwissArmyTransformer/sat/tokenization/glm/tokenization_gpt2.py:269
↓ 1 callersMethodconvert_token_to_id
(self, token)
SwissArmyTransformer/sat/tokenization/cogview/sp_tokenizer.py:127
↓ 1 callersMethodconvert_tokens_to_ids
(self, tokens)
SwissArmyTransformer/sat/tokenization/cogview/sp_tokenizer.py:98
↓ 1 callersFunctionconvert_transformer
(ckpt_path: str)
tools/convert_weight_sat2hf.py:161
↓ 1 callersFunctionconvert_vae
(ckpt_path: str)
tools/convert_weight_sat2hf.py:184
↓ 1 callersFunctioncopy_layer_norm
(src, dst)
SwissArmyTransformer/examples/deit/transform_param.py:54
↓ 1 callersFunctioncopy_layer_norm
(src, dst)
SwissArmyTransformer/examples/roberta/transform_param.py:81
↓ 1 callersFunctioncopy_layer_norm
(src, dst)
SwissArmyTransformer/examples/vit/transform_param.py:27
↓ 1 callersFunctioncopy_layer_norm
(src, dst)
SwissArmyTransformer/examples/bert/transform_param.py:81
↓ 1 callersFunctioncopy_layer_norm
(src, dst)
SwissArmyTransformer/examples/cait/transform_param.py:41
↓ 1 callersFunctioncopy_layer_norm
(src, dst)
SwissArmyTransformer/examples/eva2/transform_param.py:90
↓ 1 callersFunctioncopy_layer_norm
(src, dst)
SwissArmyTransformer/examples/yolos/transform_param.py:39
↓ 1 callersFunctioncopy_layer_norm
(src, dst)
SwissArmyTransformer/examples/chatglm/transform_param.py:73
↓ 1 callersFunctioncopy_layer_norm
(src, dst)
SwissArmyTransformer/examples/gpt2/transform_param.py:74
↓ 1 callersFunctioncopy_layer_norm
(src, dst)
SwissArmyTransformer/examples/mae/transform_param.py:42
↓ 1 callersFunctioncopy_mixin_decoder
(src, dst, ind)
SwissArmyTransformer/examples/cait/transform_param.py:86
↓ 1 callersFunctioncopy_mixin_encoder
(src, dst, ind)
SwissArmyTransformer/examples/cait/transform_param.py:67
↓ 1 callersFunctioncopy_nested_list
(src, dst)
SwissArmyTransformer/sat/model/finetune/lora2.py:38
↓ 1 callersMethodcopy_to
(self, model)
sat/sgm/modules/ema.py:52
↓ 1 callersMethodcopy_to
(self, model)
sat/vae_modules/ema.py:52
↓ 1 callersFunctioncopy_transformer_layer
(src, dst)
SwissArmyTransformer/examples/llama/transform_param.py:69
↓ 1 callersFunctioncopy_transformer_layer
(src, dst, gate)
SwissArmyTransformer/examples/mixtral-8x7b/transform_param.py:62
↓ 1 callersFunctioncopy_transformer_layer
(src, dst, w2)
SwissArmyTransformer/examples/chatglm3/transform_param.py:56
↓ 1 callersFunctioncopy_transformer_layer
(src, dst)
SwissArmyTransformer/examples/eva2clip/transform_param.py:79
↓ 1 callersFunctioncopy_transformer_layer
(src, dst, w2)
SwissArmyTransformer/examples/chatglm2/transform_param.py:53
↓ 1 callersFunctioncopy_transformer_layer
(src, dst)
SwissArmyTransformer/examples/chatglm2/transform_param_newmlp.py:54
↓ 1 callersFunctioncopy_transformer_layer_wo_ln
(src, dst)
SwissArmyTransformer/examples/deit/transform_param.py:69
↓ 1 callersFunctioncopy_transformer_layer_wo_ln
(src, dst)
SwissArmyTransformer/examples/roberta/transform_param.py:96
↓ 1 callersFunctioncopy_transformer_layer_wo_ln
(src, dst)
SwissArmyTransformer/examples/vit/transform_param.py:42
↓ 1 callersFunctioncopy_transformer_layer_wo_ln
(src, dst)
SwissArmyTransformer/examples/bert/transform_param.py:96
↓ 1 callersFunctioncopy_transformer_layer_wo_ln
(src, dst, w2)
SwissArmyTransformer/examples/eva2/transform_param.py:118
↓ 1 callersFunctioncopy_transformer_layer_wo_ln
(src, dst)
SwissArmyTransformer/examples/yolos/transform_param.py:54
↓ 1 callersFunctioncopy_transformer_layer_wo_ln
(src, dst)
SwissArmyTransformer/examples/chatglm/transform_param.py:88
↓ 1 callersFunctioncopy_transformer_layer_wo_ln
(src, dst)
SwissArmyTransformer/examples/gpt2/transform_param.py:93
↓ 1 callersFunctioncopy_transformer_layer_wo_ln_decoder
(src, dst, ind)
SwissArmyTransformer/examples/cait/transform_param.py:73
↓ 1 callersFunctioncopy_transformer_layer_wo_ln_encoder
(src, dst, ind)
SwissArmyTransformer/examples/cait/transform_param.py:56
↓ 1 callersFunctioncount_params
(model, verbose=False)
sat/sgm/util.py:247
↓ 1 callersMethodcpu_arch
(self)
SwissArmyTransformer/sat/ops/ops_builder/builder.py:318
↓ 1 callersMethoddata_iterator
iterates through data and handles wrap around
SwissArmyTransformer/sat/data_utils/samplers.py:146
↓ 1 callersMethoddecode
(self, input_ids, position_ids, attention_mask, encoder_outputs,cross_attention_mask=None, **kw_args)
SwissArmyTransformer/sat/model/encoder_decoder_model.py:67
↓ 1 callersMethoddecode
(self, input_ids, attention_mask=None, encoder_outputs=None, cross_attention_mask=None, **kw_args)
SwissArmyTransformer/sat/model/official/t5_model.py:270
↓ 1 callersMethoddecode
( self, z: torch.Tensor, input_cp: bool = False, output_cp: bool = False,
sat/sgm/models/autoencoder.py:594
↓ 1 callersMethoddecode
( self, z: torch.Tensor, input_cp: bool = False, output_cp: bool = False,
sat/vae_modules/autoencoder.py:617
↓ 1 callersMethoddecode_code
(self, code_t)
SwissArmyTransformer/sat/tokenization/cogview/vqvae/vqvae_zc.py:271
↓ 1 callersFunctiondecode_latents
(model, latents)
sat/sample_video.py:115
↓ 1 callersMethoddecompose
(self)
SwissArmyTransformer/examples/yolos/util/misc.py:299
↓ 1 callersMethoddisable_untrainable_params
(self)
SwissArmyTransformer/sat/model/base_model.py:178
↓ 1 callersFunctiondivisible_by
(num, den)
sat/sgm/modules/cp_enc_dec.py:32
↓ 1 callersFunctiondivisible_by
(num, den)
sat/sgm/modules/autoencoding/vqvae/movq_dec_3d.py:14
↓ 1 callersFunctiondivisible_by
(num, den)
sat/sgm/modules/autoencoding/vqvae/movq_enc_3d.py:17
↓ 1 callersFunctiondivisible_by
(num, den)
sat/sgm/modules/autoencoding/vqvae/movq_dec_3d_dev.py:19
↓ 1 callersFunctiondivisible_by
(num, den)
sat/vae_modules/cp_enc_dec.py:30
↓ 1 callersFunctiondownload
(url, local_path, chunk_size=1024)
sat/sgm/modules/autoencoding/lpips/util.py:16
↓ 1 callersFunctiondownload_s3
Download remote_dir into (under) local_dir
SwissArmyTransformer/sat/resources/download.py:84
← previousnext →601–700 of 2,541, ranked by callers