MCPcopy Create free account

hub / github.com/ByteDance-Seed/Bagel / types & classes

Types & classes116 in github.com/ByteDance-Seed/Bagel

↓ 21 callersClassNaiveCache
modeling/bagel/qwen2_navit.py:207
↓ 19 callersClassQwen2RMSNorm
modeling/qwen2/modeling_qwen2.py:45
↓ 13 callersClassImageTransform
data/transforms.py:90
↓ 9 callersClassBagel
modeling/bagel/bagel.py:57
↓ 9 callersClassBagelConfig
modeling/bagel/bagel.py:27
↓ 9 callersClassQwen2ForCausalLM
modeling/bagel/qwen2_navit.py:1095
↓ 9 callersClassSiglipVisionModel
modeling/bagel/siglip_navit.py:374
↓ 6 callersClassQwen2MLP
modeling/qwen2/modeling_qwen2.py:190
↓ 6 callersClassResnetBlock
modeling/autoencoder.py:68
↓ 2 callersClassAttnBlock
modeling/autoencoder.py:38
↓ 2 callersClassEvalAIAnswerProcessor
Processes an answer similar to Eval AI copied from https://github.com/facebookresearch/mmf/blob/c46b3b3391275b4181567db80943473a8
eval/vlm/eval/vqa/textvqa_eval.py:17
↓ 2 callersClassGPT4o
eval/gen/gedit/viescore/mllm_tools/openai.py:180
↓ 2 callersClassPackedAttention
modeling/bagel/qwen2_navit.py:236
↓ 2 callersClassPositionEmbedding
modeling/bagel/modeling_utils.py:127
↓ 2 callersClassQwen25VL
eval/gen/gedit/viescore/mllm_tools/qwen25vl_eval.py:42
↓ 2 callersClassQwen2RotaryEmbedding
modeling/qwen2/modeling_qwen2.py:66
↓ 2 callersClassSiglipEncoder
Transformer encoder consisting of `config.num_hidden_layers` self attention layers. Each layer is a [`SiglipEncoderLayer`]. Args:
modeling/siglip/modeling_siglip.py:824
↓ 2 callersClassSiglipMLP
modeling/siglip/modeling_siglip.py:588
↓ 2 callersClassVIEScore
eval/gen/gedit/viescore/__init__.py:10
↓ 1 callersClassAutoEncoder
modeling/autoencoder.py:290
↓ 1 callersClassAutoEncoderParams
modeling/autoencoder.py:21
↓ 1 callersClassBaseNavitOutputWithPast
modeling/bagel/qwen2_navit.py:225
↓ 1 callersClassDataConfig
data/dataset_base.py:23
↓ 1 callersClassDecoder
modeling/autoencoder.py:196
↓ 1 callersClassDiagonalGaussian
modeling/autoencoder.py:275
↓ 1 callersClassDownsample
modeling/autoencoder.py:98
↓ 1 callersClassEncoder
modeling/autoencoder.py:122
↓ 1 callersClassFSDPConfig
train/fsdp_utils.py:32
↓ 1 callersClassFrameSampler
data/video_utils.py:117
↓ 1 callersClassImageCrops
eval/gen/geneval/evaluation/evaluate_images.py:93
↓ 1 callersClassImageCrops
eval/gen/geneval/evaluation/evaluate_images_mp.py:97
↓ 1 callersClassInferenceSampler
eval/vlm/eval/vqa/evaluate_vqa.py:274
↓ 1 callersClassInferenceSampler
eval/vlm/eval/mmbench/evaluate_mmbench.py:138
↓ 1 callersClassInferenceSampler
eval/vlm/eval/pope/evaluate_pope.py:107
↓ 1 callersClassInferenceSampler
eval/vlm/eval/mmvp/evaluate_mmvp.py:98
↓ 1 callersClassInferenceSampler
eval/vlm/eval/mathvista/evaluate_mathvista.py:78
↓ 1 callersClassInferenceSampler
eval/vlm/eval/mmmu/evaluate_mmmu.py:118
↓ 1 callersClassInferenceSampler
eval/vlm/eval/mmmu/evaluate_mmmu_cot.py:142
↓ 1 callersClassInterleaveInferencer
inferencer.py:22
↓ 1 callersClassMLPconnector
modeling/bagel/modeling_utils.py:113
↓ 1 callersClassMMBenchDataset
eval/vlm/eval/mmbench/evaluate_mmbench.py:82
↓ 1 callersClassMMMUDataset
eval/vlm/eval/mmmu/evaluate_mmmu.py:57
↓ 1 callersClassMMMUDataset
eval/vlm/eval/mmmu/evaluate_mmmu_cot.py:82
↓ 1 callersClassMMVPDataset
eval/vlm/eval/mmvp/evaluate_mmvp.py:43
↓ 1 callersClassMathVistaDataset
eval/vlm/eval/mathvista/evaluate_mathvista.py:56
↓ 1 callersClassMaxLongEdgeMinShortEdgeResize
Resize the input image so that its longest side and shortest side are within a specified range, ensuring that both sides are divisible by a specif
data/transforms.py:15
↓ 1 callersClassPackedDataset
data/dataset_base.py:45
↓ 1 callersClassQwen2DecoderLayer
modeling/qwen2/modeling_qwen2.py:447
↓ 1 callersClassQwen2Model
Transformer decoder consisting of *config.num_hidden_layers* layers. Each layer is a [`Qwen2DecoderLayer`] Args: config: Qwen2Config
modeling/qwen2/modeling_qwen2.py:654
↓ 1 callersClassQwen2Model
modeling/bagel/qwen2_navit.py:943
↓ 1 callersClassRotaryEmbedding2D
modeling/bagel/siglip_navit.py:102
↓ 1 callersClassSiglipConfig
r""" [`SiglipConfig`] is the configuration class to store the configuration of a [`SiglipModel`]. It is used to instantiate a Siglip model acc
modeling/siglip/configuration_siglip.py:217
↓ 1 callersClassSiglipEncoder
modeling/bagel/siglip_navit.py:303
↓ 1 callersClassSiglipEncoderLayer
modeling/bagel/siglip_navit.py:262
↓ 1 callersClassSiglipEncoderLayer
modeling/siglip/modeling_siglip.py:603
↓ 1 callersClassSiglipFlashAttention2
modeling/bagel/siglip_navit.py:198
↓ 1 callersClassSiglipImageProcessor
r""" Constructs a SigLIP image processor. Args: do_resize (`bool`, *optional*, defaults to `True`): Whether to resize the
modeling/siglip/image_processing_siglip.py:37
↓ 1 callersClassSiglipMLP
modeling/bagel/siglip_navit.py:247
↓ 1 callersClassSiglipModel
modeling/siglip/modeling_siglip.py:1189
↓ 1 callersClassSiglipMultiheadAttentionPoolingHead
Multihead Attention Pooling.
modeling/siglip/modeling_siglip.py:1102
↓ 1 callersClassSiglipOutput
Args: loss (`torch.FloatTensor` of shape `(1,)`, *optional*, returned when `return_loss` is `True`): Contrastive loss for ima
modeling/siglip/modeling_siglip.py:203
↓ 1 callersClassSiglipProcessor
r""" Constructs a Siglip processor which wraps a Siglip image processor and a Siglip tokenizer into a single processor. [`SiglipProcessor`] o
modeling/siglip/processing_siglip.py:17
↓ 1 callersClassSiglipTextConfig
r""" This is the configuration class to store the configuration of a [`SiglipTextModel`]. It is used to instantiate a Siglip text encoder acco
modeling/siglip/configuration_siglip.py:16
↓ 1 callersClassSiglipTextEmbeddings
modeling/siglip/modeling_siglip.py:311
↓ 1 callersClassSiglipTextTransformer
modeling/siglip/modeling_siglip.py:912
↓ 1 callersClassSiglipTokenizer
Construct a Siglip tokenizer. Based on [SentencePiece](https://github.com/google/sentencepiece). This tokenizer inherits from [`PreTrainedTo
modeling/siglip/tokenization_siglip.py:33
↓ 1 callersClassSiglipVisionConfig
r""" This is the configuration class to store the configuration of a [`SiglipVisionModel`]. It is used to instantiate a Siglip vision encoder
modeling/siglip/configuration_siglip.py:121
↓ 1 callersClassSiglipVisionEmbeddings
modeling/bagel/siglip_navit.py:145
↓ 1 callersClassSiglipVisionEmbeddings
modeling/siglip/modeling_siglip.py:239
↓ 1 callersClassSiglipVisionTransformer
modeling/bagel/siglip_navit.py:330
↓ 1 callersClassSiglipVisionTransformer
modeling/siglip/modeling_siglip.py:1045
↓ 1 callersClassSimpleCustomBatch
data/dataset_base.py:478
↓ 1 callersClassTextVQAAccuracyEvaluator
eval/vlm/eval/vqa/textvqa_eval.py:231
↓ 1 callersClassTimestepEmbedder
Embeds scalar timesteps into vector representations.
modeling/bagel/modeling_utils.py:74
↓ 1 callersClassUpsample
modeling/autoencoder.py:111
↓ 1 callersClassVQADataset
eval/vlm/eval/vqa/evaluate_vqa.py:231
↓ 1 callersClassVQADataset
eval/vlm/eval/pope/evaluate_pope.py:70
↓ 1 callersClassVQADataset
eval/vlm/eval/mmvet/evaluate_mmvet.py:33
↓ 1 callersClasscalculate_metrics
eval/vlm/eval/mme/calculation.py:28
ClassDataArguments
train/pretrain_unified_navit.py:176
ClassDistributedIterableDataset
data/distributed_iterable_dataset.py:8
ClassFSDPCheckpoint
train/fsdp_utils.py:86
ClassGPT4v
eval/gen/gedit/viescore/mllm_tools/openai.py:80
ClassInterleavedBaseIterableDataset
data/interleave_datasets/interleave_t2i_dataset.py:10
ClassModelArguments
train/pretrain_unified_navit.py:99
ClassNumpyEncoder
eval/gen/rise/utils.py:13
ClassPackedAttentionMoT
modeling/bagel/qwen2_navit.py:381
ClassParquetStandardIterableDataset
data/interleave_datasets/interleave_t2i_dataset.py:132
ClassQwen2Attention
Multi-headed attention from 'Attention Is All You Need' paper. Modified to use sliding window attention: Longformer and "Generating Long Sequ
modeling/qwen2/modeling_qwen2.py:217
ClassQwen2Config
r""" This is the configuration class to store the configuration of a [`Qwen2Model`]. It is used to instantiate a Qwen2 model according to the
modeling/qwen2/configuration_qwen2.py:14
ClassQwen2Config
r""" This is the configuration class to store the configuration of a [`Qwen2Model`]. It is used to instantiate a Qwen2 model according to the
modeling/bagel/qwen2_navit.py:46
ClassQwen2DecoderLayer
modeling/bagel/qwen2_navit.py:603
ClassQwen2FlashAttention2
Qwen2 flash attention module, following Qwen2 attention module. This module inherits from `Qwen2Attention` as the weights of the module stays
modeling/qwen2/modeling_qwen2.py:322
ClassQwen2ForCausalLM
modeling/qwen2/modeling_qwen2.py:812
ClassQwen2MoEDecoderLayer
modeling/bagel/qwen2_navit.py:834
ClassQwen2MoTDecoderLayer
modeling/bagel/qwen2_navit.py:687
ClassQwen2PreTrainedModel
modeling/qwen2/modeling_qwen2.py:552
ClassQwen2Tokenizer
Construct a Qwen2 tokenizer. Based on byte-level Byte-Pair-Encoding. Same with GPT2Tokenizer, this tokenizer has been trained to treat space
modeling/qwen2/tokenization_qwen2.py:72
ClassQwen2TokenizerFast
Construct a "fast" Qwen2 tokenizer (backed by HuggingFace's *tokenizers* library). Based on byte-level Byte-Pair-Encoding. Same with GPT
modeling/qwen2/tokenization_qwen2_fast.py:26
ClassSTVQAANLSEvaluator
eval/vlm/eval/vqa/textvqa_eval.py:286
next →1–100 of 116, ranked by callers