Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/ByteDance-Seed/Bagel
/ types & classes
Types & classes
116 in github.com/ByteDance-Seed/Bagel
⨍
Functions
720
◇
Types & classes
116
↳
Endpoints
2
↓ 21 callers
Class
NaiveCache
modeling/bagel/qwen2_navit.py:207
↓ 19 callers
Class
Qwen2RMSNorm
modeling/qwen2/modeling_qwen2.py:45
↓ 13 callers
Class
ImageTransform
data/transforms.py:90
↓ 9 callers
Class
Bagel
modeling/bagel/bagel.py:57
↓ 9 callers
Class
BagelConfig
modeling/bagel/bagel.py:27
↓ 9 callers
Class
Qwen2ForCausalLM
modeling/bagel/qwen2_navit.py:1095
↓ 9 callers
Class
SiglipVisionModel
modeling/bagel/siglip_navit.py:374
↓ 6 callers
Class
Qwen2MLP
modeling/qwen2/modeling_qwen2.py:190
↓ 6 callers
Class
ResnetBlock
modeling/autoencoder.py:68
↓ 2 callers
Class
AttnBlock
modeling/autoencoder.py:38
↓ 2 callers
Class
EvalAIAnswerProcessor
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 callers
Class
GPT4o
eval/gen/gedit/viescore/mllm_tools/openai.py:180
↓ 2 callers
Class
PackedAttention
modeling/bagel/qwen2_navit.py:236
↓ 2 callers
Class
PositionEmbedding
modeling/bagel/modeling_utils.py:127
↓ 2 callers
Class
Qwen25VL
eval/gen/gedit/viescore/mllm_tools/qwen25vl_eval.py:42
↓ 2 callers
Class
Qwen2RotaryEmbedding
modeling/qwen2/modeling_qwen2.py:66
↓ 2 callers
Class
SiglipEncoder
Transformer encoder consisting of `config.num_hidden_layers` self attention layers. Each layer is a [`SiglipEncoderLayer`]. Args:
modeling/siglip/modeling_siglip.py:824
↓ 2 callers
Class
SiglipMLP
modeling/siglip/modeling_siglip.py:588
↓ 2 callers
Class
VIEScore
eval/gen/gedit/viescore/__init__.py:10
↓ 1 callers
Class
AutoEncoder
modeling/autoencoder.py:290
↓ 1 callers
Class
AutoEncoderParams
modeling/autoencoder.py:21
↓ 1 callers
Class
BaseNavitOutputWithPast
modeling/bagel/qwen2_navit.py:225
↓ 1 callers
Class
DataConfig
data/dataset_base.py:23
↓ 1 callers
Class
Decoder
modeling/autoencoder.py:196
↓ 1 callers
Class
DiagonalGaussian
modeling/autoencoder.py:275
↓ 1 callers
Class
Downsample
modeling/autoencoder.py:98
↓ 1 callers
Class
Encoder
modeling/autoencoder.py:122
↓ 1 callers
Class
FSDPConfig
train/fsdp_utils.py:32
↓ 1 callers
Class
FrameSampler
data/video_utils.py:117
↓ 1 callers
Class
ImageCrops
eval/gen/geneval/evaluation/evaluate_images.py:93
↓ 1 callers
Class
ImageCrops
eval/gen/geneval/evaluation/evaluate_images_mp.py:97
↓ 1 callers
Class
InferenceSampler
eval/vlm/eval/vqa/evaluate_vqa.py:274
↓ 1 callers
Class
InferenceSampler
eval/vlm/eval/mmbench/evaluate_mmbench.py:138
↓ 1 callers
Class
InferenceSampler
eval/vlm/eval/pope/evaluate_pope.py:107
↓ 1 callers
Class
InferenceSampler
eval/vlm/eval/mmvp/evaluate_mmvp.py:98
↓ 1 callers
Class
InferenceSampler
eval/vlm/eval/mathvista/evaluate_mathvista.py:78
↓ 1 callers
Class
InferenceSampler
eval/vlm/eval/mmmu/evaluate_mmmu.py:118
↓ 1 callers
Class
InferenceSampler
eval/vlm/eval/mmmu/evaluate_mmmu_cot.py:142
↓ 1 callers
Class
InterleaveInferencer
inferencer.py:22
↓ 1 callers
Class
MLPconnector
modeling/bagel/modeling_utils.py:113
↓ 1 callers
Class
MMBenchDataset
eval/vlm/eval/mmbench/evaluate_mmbench.py:82
↓ 1 callers
Class
MMMUDataset
eval/vlm/eval/mmmu/evaluate_mmmu.py:57
↓ 1 callers
Class
MMMUDataset
eval/vlm/eval/mmmu/evaluate_mmmu_cot.py:82
↓ 1 callers
Class
MMVPDataset
eval/vlm/eval/mmvp/evaluate_mmvp.py:43
↓ 1 callers
Class
MathVistaDataset
eval/vlm/eval/mathvista/evaluate_mathvista.py:56
↓ 1 callers
Class
MaxLongEdgeMinShortEdgeResize
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 callers
Class
PackedDataset
data/dataset_base.py:45
↓ 1 callers
Class
Qwen2DecoderLayer
modeling/qwen2/modeling_qwen2.py:447
↓ 1 callers
Class
Qwen2Model
Transformer decoder consisting of *config.num_hidden_layers* layers. Each layer is a [`Qwen2DecoderLayer`] Args: config: Qwen2Config
modeling/qwen2/modeling_qwen2.py:654
↓ 1 callers
Class
Qwen2Model
modeling/bagel/qwen2_navit.py:943
↓ 1 callers
Class
RotaryEmbedding2D
modeling/bagel/siglip_navit.py:102
↓ 1 callers
Class
SiglipConfig
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 callers
Class
SiglipEncoder
modeling/bagel/siglip_navit.py:303
↓ 1 callers
Class
SiglipEncoderLayer
modeling/bagel/siglip_navit.py:262
↓ 1 callers
Class
SiglipEncoderLayer
modeling/siglip/modeling_siglip.py:603
↓ 1 callers
Class
SiglipFlashAttention2
modeling/bagel/siglip_navit.py:198
↓ 1 callers
Class
SiglipImageProcessor
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 callers
Class
SiglipMLP
modeling/bagel/siglip_navit.py:247
↓ 1 callers
Class
SiglipModel
modeling/siglip/modeling_siglip.py:1189
↓ 1 callers
Class
SiglipMultiheadAttentionPoolingHead
Multihead Attention Pooling.
modeling/siglip/modeling_siglip.py:1102
↓ 1 callers
Class
SiglipOutput
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 callers
Class
SiglipProcessor
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 callers
Class
SiglipTextConfig
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 callers
Class
SiglipTextEmbeddings
modeling/siglip/modeling_siglip.py:311
↓ 1 callers
Class
SiglipTextTransformer
modeling/siglip/modeling_siglip.py:912
↓ 1 callers
Class
SiglipTokenizer
Construct a Siglip tokenizer. Based on [SentencePiece](https://github.com/google/sentencepiece). This tokenizer inherits from [`PreTrainedTo
modeling/siglip/tokenization_siglip.py:33
↓ 1 callers
Class
SiglipVisionConfig
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 callers
Class
SiglipVisionEmbeddings
modeling/bagel/siglip_navit.py:145
↓ 1 callers
Class
SiglipVisionEmbeddings
modeling/siglip/modeling_siglip.py:239
↓ 1 callers
Class
SiglipVisionTransformer
modeling/bagel/siglip_navit.py:330
↓ 1 callers
Class
SiglipVisionTransformer
modeling/siglip/modeling_siglip.py:1045
↓ 1 callers
Class
SimpleCustomBatch
data/dataset_base.py:478
↓ 1 callers
Class
TextVQAAccuracyEvaluator
eval/vlm/eval/vqa/textvqa_eval.py:231
↓ 1 callers
Class
TimestepEmbedder
Embeds scalar timesteps into vector representations.
modeling/bagel/modeling_utils.py:74
↓ 1 callers
Class
Upsample
modeling/autoencoder.py:111
↓ 1 callers
Class
VQADataset
eval/vlm/eval/vqa/evaluate_vqa.py:231
↓ 1 callers
Class
VQADataset
eval/vlm/eval/pope/evaluate_pope.py:70
↓ 1 callers
Class
VQADataset
eval/vlm/eval/mmvet/evaluate_mmvet.py:33
↓ 1 callers
Class
calculate_metrics
eval/vlm/eval/mme/calculation.py:28
Class
DataArguments
train/pretrain_unified_navit.py:176
Class
DistributedIterableDataset
data/distributed_iterable_dataset.py:8
Class
FSDPCheckpoint
train/fsdp_utils.py:86
Class
GPT4v
eval/gen/gedit/viescore/mllm_tools/openai.py:80
Class
InterleavedBaseIterableDataset
data/interleave_datasets/interleave_t2i_dataset.py:10
Class
ModelArguments
train/pretrain_unified_navit.py:99
Class
NumpyEncoder
eval/gen/rise/utils.py:13
Class
PackedAttentionMoT
modeling/bagel/qwen2_navit.py:381
Class
ParquetStandardIterableDataset
data/interleave_datasets/interleave_t2i_dataset.py:132
Class
Qwen2Attention
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
Class
Qwen2Config
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
Class
Qwen2Config
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
Class
Qwen2DecoderLayer
modeling/bagel/qwen2_navit.py:603
Class
Qwen2FlashAttention2
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
Class
Qwen2ForCausalLM
modeling/qwen2/modeling_qwen2.py:812
Class
Qwen2MoEDecoderLayer
modeling/bagel/qwen2_navit.py:834
Class
Qwen2MoTDecoderLayer
modeling/bagel/qwen2_navit.py:687
Class
Qwen2PreTrainedModel
modeling/qwen2/modeling_qwen2.py:552
Class
Qwen2Tokenizer
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
Class
Qwen2TokenizerFast
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
Class
STVQAANLSEvaluator
eval/vlm/eval/vqa/textvqa_eval.py:286
next →
1–100 of 116, ranked by callers