MCPcopy Create free account

hub / github.com/baaivision/Emu3 / types & classes

Types & classes39 in github.com/baaivision/Emu3

↓ 7 callersClassEmu3VisionVQCausalConv3d
emu3/tokenizer/modeling_emu3visionvq.py:74
↓ 6 callersClassEmu3VisionVQResnetBlock
emu3/tokenizer/modeling_emu3visionvq.py:236
↓ 5 callersClassEmu3Processor
r""" Constructs an Emu3 processor which wraps an Emu3 image processor and an Emu3 vision vq model and an Emu3 tokenizer into a single processor.
emu3/mllm/processing_emu3.py:37
↓ 4 callersClassEmu3VisionVQActivation
emu3/tokenizer/modeling_emu3visionvq.py:28
↓ 4 callersClassEmu3VisionVQAttnBlock
emu3/tokenizer/modeling_emu3visionvq.py:320
↓ 4 callersClassEmu3VisionVQSpatialNorm
emu3/tokenizer/modeling_emu3visionvq.py:180
↓ 3 callersClassEmu3RMSNorm
emu3/mllm/modeling_emu3.py:106
↓ 2 callersClassEmu3VisionVQResnetTemporalBlock
emu3/tokenizer/modeling_emu3visionvq.py:109
↓ 1 callersClassEmu3DecoderLayer
emu3/mllm/modeling_emu3.py:754
↓ 1 callersClassEmu3DynamicNTKScalingRotaryEmbedding
Emu3RotaryEmbedding extended with Dynamic NTK scaling. Credits to the Reddit users /u/bloc97 and /u/emozilla
emu3/mllm/modeling_emu3.py:181
↓ 1 callersClassEmu3FeatureDataset
emu3/train/datasets.py:11
↓ 1 callersClassEmu3LinearScalingRotaryEmbedding
Emu3RotaryEmbedding extended with linear scaling. Credits to the Reddit user /u/kaiokendev
emu3/mllm/modeling_emu3.py:162
↓ 1 callersClassEmu3MLP
emu3/mllm/modeling_emu3.py:242
↓ 1 callersClassEmu3Model
Transformer decoder consisting of *config.num_hidden_layers* layers. Each layer is a [`Emu3DecoderLayer`] Args: config: Emu3Config
emu3/mllm/modeling_emu3.py:944
↓ 1 callersClassEmu3RotaryEmbedding
emu3/mllm/modeling_emu3.py:126
↓ 1 callersClassEmu3VisionVQDecoder
emu3/tokenizer/modeling_emu3visionvq.py:596
↓ 1 callersClassEmu3VisionVQDownsample
emu3/tokenizer/modeling_emu3visionvq.py:55
↓ 1 callersClassEmu3VisionVQEncoder
emu3/tokenizer/modeling_emu3visionvq.py:471
↓ 1 callersClassEmu3VisionVQTemporalDownsample
emu3/tokenizer/modeling_emu3visionvq.py:421
↓ 1 callersClassEmu3VisionVQTemporalUpsample
emu3/tokenizer/modeling_emu3visionvq.py:393
↓ 1 callersClassEmu3VisionVQUpsample
emu3/tokenizer/modeling_emu3visionvq.py:37
↓ 1 callersClassEmu3VisionVQVectorQuantizer
emu3/tokenizer/modeling_emu3visionvq.py:447
ClassDataArguments
emu3/train/train.py:22
ClassEmu3Attention
Multi-headed attention from 'Attention Is All You Need' paper
emu3/mllm/modeling_emu3.py:288
ClassEmu3Config
r""" This is the configuration class to store the configuration of a [`Emu3Model`]. It is used to instantiate an Emu3 model according to the s
emu3/mllm/configuration_emu3.py:33
ClassEmu3FlashAttention2
Emu3 flash attention module. This module inherits from `Emu3Attention` as the weights of the module stays untouched. The only required change
emu3/mllm/modeling_emu3.py:459
ClassEmu3ForCausalLM
emu3/mllm/modeling_emu3.py:1108
ClassEmu3PreTrainedModel
emu3/mllm/modeling_emu3.py:848
ClassEmu3PrefixConstrainedLogitsHelper
emu3/mllm/utils_emu3.py:19
ClassEmu3SdpaAttention
Emu3 attention module using torch.nn.functional.scaled_dot_product_attention. This module inherits from `Emu3Attention` as the weights of the
emu3/mllm/modeling_emu3.py:660
ClassEmu3Tokenizer
Emu3 tokenizer.
emu3/mllm/tokenization_emu3.py:55
ClassEmu3VisionVQConfig
r""" This is the configuration class to store the configuration of a [`Emu3VisionVQ`]. It is used to instantiate an video movq model accordin
emu3/tokenizer/configuration_emu3visionvq.py:26
ClassEmu3VisionVQImageProcessor
r""" Constructs a Emu3VisionVQ image processor that dynamically resizes images based on the original images. Args: do_resize (`bool`,
emu3/tokenizer/image_processing_emu3visionvq.py:86
ClassEmu3VisionVQModel
emu3/tokenizer/modeling_emu3visionvq.py:749
ClassEmu3VisionVQPretrainedModel
An abstract class to handle weights initialization and a simple interface for downloading and loading pretrained models.
emu3/tokenizer/modeling_emu3visionvq.py:723
ClassModelArguments
emu3/train/train.py:17
ClassPredictor
replicate_demo/predict_gen.py:50
ClassPredictor
replicate_demo/predict_chat.py:45
ClassTrainingArguments
emu3/train/train.py:33