MCPcopy Create free account

hub / github.com/apple/ml-fastvlm / functions

Functions294 in github.com/apple/ml-fastvlm

↓ 1 callersMethodget_kernel_bias
Method to obtain re-parameterized kernel and bias. Reference: https://github.com/DingXiaoH/RepLKNet-pytorch Returns: Tupl
llava/model/multimodal_encoder/mobileclip/mci.py:453
↓ 1 callersFunctionget_mm_adapter_state_maybe_zero_3
(named_params, keys_to_match)
llava/train/train.py:171
↓ 1 callersFunctionget_mm_adapter_state_maybe_zero_3
(named_params, keys_to_match)
llava/train/llava_trainer.py:36
↓ 1 callersFunctionget_mm_adapter_state_maybe_zero_3
(named_params, keys_to_match)
llava/train/train_qwen.py:171
↓ 1 callersFunctionget_modality_length_grouped_indices
(lengths, batch_size, world_size, generator=None)
llava/train/llava_trainer.py:64
↓ 1 callersMethodget_model
(self)
llava/model/language_model/llava_qwen.py:54
↓ 1 callersMethodget_model
(self)
llava/model/language_model/llava_llama.py:54
↓ 1 callersMethodget_model
(self)
llava/model/language_model/llava_mistral.py:54
↓ 1 callersFunctionget_peft_state_maybe_zero_3
(named_params, bias)
llava/train/train.py:138
↓ 1 callersFunctionget_peft_state_maybe_zero_3
(named_params, bias)
llava/train/train_qwen.py:138
↓ 1 callersFunctionget_peft_state_non_lora_maybe_zero_3
(named_params, require_grad_only=True)
llava/train/train.py:163
↓ 1 callersFunctionget_peft_state_non_lora_maybe_zero_3
(named_params, require_grad_only=True)
llava/train/train_qwen.py:163
↓ 1 callersFunctionget_tokenize_len
(prompts)
llava/train/train.py:746
↓ 1 callersFunctionget_tokenize_len
(prompts)
llava/train/train_qwen.py:870
↓ 1 callersFunctioninsert_separator
(X, sep)
llava/mm_utils.py:190
↓ 1 callersMethodlist_models
(self)
llava/serve/controller.py:112
↓ 1 callersFunctionload_from_hf
(repo_id, filename, subfolder=None)
llava/model/builder.py:70
↓ 1 callersFunctionload_image
(image_file)
llava/serve/cli.py:18
↓ 1 callersFunctionmain
(args)
llava/serve/cli.py:27
↓ 1 callersFunctionmain
()
llava/serve/test_message.py:9
↓ 1 callersFunctionmake_delta
(base_model_path, target_model_path, delta_path, hub_repo_id)
llava/model/make_delta.py:13
↓ 1 callersFunctionmake_supervised_data_module
Make dataset and collator for supervised fine-tuning.
llava/train/train.py:897
↓ 1 callersFunctionmake_supervised_data_module
Make dataset and collator for supervised fine-tuning.
llava/train/train_qwen.py:1029
↓ 1 callersFunctionmaybe_zero_3
(param, ignore_status=False, name=None)
llava/train/llava_trainer.py:22
↓ 1 callersMethodpool
(self, x)
llava/model/multimodal_encoder/mobileclip/mci.py:1282
↓ 1 callersFunctionpredict
(args)
predict.py:18
↓ 1 callersFunctionpreprocess
Given a list of sources, each is a conversation list. This transform: 1. Add signal '### ' at the beginning each sentence, with end signal '\
llava/train/train.py:715
↓ 1 callersFunctionpreprocess
Given a list of sources, each is a conversation list. This transform: 1. Add signal '### ' at the beginning each sentence, with end signal '\
llava/train/train_qwen.py:833
↓ 1 callersFunctionpreprocess_llama_2
( sources, tokenizer: transformers.PreTrainedTokenizer, has_image: bool = False )
llava/train/train.py:340
↓ 1 callersFunctionpreprocess_llama_2
( sources, tokenizer: transformers.PreTrainedTokenizer, has_image: bool = False )
llava/train/train_qwen.py:340
↓ 1 callersFunctionpreprocess_mpt
( sources, tokenizer: transformers.PreTrainedTokenizer, has_image: bool = False )
llava/train/train.py:605
↓ 1 callersFunctionpreprocess_mpt
( sources, tokenizer: transformers.PreTrainedTokenizer, has_image: bool = False )
llava/train/train_qwen.py:723
↓ 1 callersFunctionpreprocess_multimodal
( sources: Sequence[str], data_args: DataArguments )
llava/train/train.py:316
↓ 1 callersFunctionpreprocess_multimodal
( sources: Sequence[str], data_args: DataArguments )
llava/train/train_qwen.py:316
↓ 1 callersFunctionpreprocess_plain
( sources: Sequence[str], tokenizer: transformers.PreTrainedTokenizer, )
llava/train/train.py:693
↓ 1 callersFunctionpreprocess_plain
( sources: Sequence[str], tokenizer: transformers.PreTrainedTokenizer, )
llava/train/train_qwen.py:811
↓ 1 callersFunctionpreprocess_qwen_2
( sources, tokenizer: transformers.PreTrainedTokenizer, has_image: bool = False )
llava/train/train.py:423
↓ 1 callersFunctionpreprocess_qwen_2
( sources, tokenizer: transformers.PreTrainedTokenizer, has_image: bool = False )
llava/train/train_qwen.py:527
↓ 1 callersFunctionpreprocess_v1
( sources, tokenizer: transformers.PreTrainedTokenizer, has_image: bool = False )
llava/train/train.py:519
↓ 1 callersFunctionpreprocess_v1
( sources, tokenizer: transformers.PreTrainedTokenizer, has_image: bool = False )
llava/train/train_qwen.py:637
↓ 1 callersFunctionrank0_print
(*args)
llava/train/train_qwen.py:45
↓ 1 callersMethodreceive_heart_beat
(self, worker_name: str, queue_length: int)
llava/serve/controller.py:173
↓ 1 callersMethodrefresh_all_workers
(self)
llava/serve/controller.py:104
↓ 1 callersMethodremove_stable_workers_by_expiration
(self)
llava/serve/controller.py:183
↓ 1 callersFunctionreplace_llama_attn_with_xformers_attn
()
llava/train/llama_xformers_attn_monkey_patch.py:19
↓ 1 callersFunctionresize_and_pad_image
Resize and pad an image to a target resolution while maintaining aspect ratio. Args: image (PIL.Image.Image): The input image.
llava/mm_utils.py:44
↓ 1 callersFunctionsafe_save_model_for_hf_trainer
Collects the state dict and dump to disk.
llava/train/train.py:193
↓ 1 callersFunctionsafe_save_model_for_hf_trainer
Collects the state dict and dump to disk.
llava/train/train_qwen.py:193
↓ 1 callersFunctionsmart_tokenizer_and_embedding_resize
Resize tokenizer and embedding. Note: This is the unoptimized version that may make your embedding size not be divisible by 64.
llava/train/train.py:232
↓ 1 callersFunctionsmart_tokenizer_and_embedding_resize
Resize tokenizer and embedding. Note: This is the unoptimized version that may make your embedding size not be divisible by 64.
llava/train/train_qwen.py:232
↓ 1 callersFunctionsplit_to_even_chunks
Split a list of indices into `chunks` chunks of roughly equal lengths.
llava/train/llava_trainer.py:42
↓ 1 callersFunctionunpad_image
Unpads a PyTorch tensor of a padded and resized image. Args: tensor (torch.Tensor): The image tensor, assumed to be in CxHxW format.
llava/model/llava_arch.py:101
↓ 1 callersFunctionviolates_moderation
Check whether the text violates OpenAI moderation API.
llava/utils.py:103
↓ 1 callersMethodworker_api_generate_stream
(self, params)
llava/serve/controller.py:193
↓ 1 callersMethodworker_api_get_status
(self)
llava/serve/controller.py:220
Method__call__
(self, output_ids: torch.LongTensor, scores: torch.FloatTensor, **kwargs)
llava/mm_utils.py:246
Method__call__
(self, instances: Sequence[Dict])
llava/train/train.py:868
Method__call__
(self, instances: Sequence[Dict])
llava/train/train_qwen.py:1000
Method__getattr__
(self, attr)
llava/utils.py:71
Method__getitem__
(self, i)
llava/train/train.py:805
Method__getitem__
(self, i)
llava/train/train_qwen.py:985
Method__init__
(self, logger, log_level=logging.INFO)
llava/utils.py:65
Method__init__
(self, keywords, tokenizer, input_ids)
llava/mm_utils.py:219
Method__init__
(self, controller_addr, worker_addr, worker_id, no_register, model_path, mod
llava/serve/model_worker.py:45
Method__init__
(self, controller_addr, worker_addr, sgl_endpoint, worker_id, no_register, model_name)
llava/serve/sglang_worker.py:55
Method__init__
(self, dispatch_method: str)
llava/serve/controller.py:58
Method__init__
(self, config)
llava/model/llava_arch.py:31
Method__init__
(self, config)
llava/model/language_model/llava_qwen.py:44
Method__init__
(self, config)
llava/model/language_model/llava_llama.py:44
Method__init__
(self, config)
llava/model/language_model/llava_mpt.py:44
Method__init__
(self, config)
llava/model/language_model/llava_mistral.py:45
Method__init__
(self)
llava/model/multimodal_projector/builder.py:6
Method__init__
(self, vision_tower, args, delay_load=False)
llava/model/multimodal_encoder/mobileclip_encoder.py:14
Method__init__
(self, vision_tower, args, delay_load=False)
llava/model/multimodal_encoder/clip_encoder.py:112
Method__init__
(self, model_name: str, *args, **kwargs)
llava/model/multimodal_encoder/mobileclip/__init__.py:39
Method__init__
Construct a Squeeze and Excite Module. Args: in_channels: Number of input channels. rd_ratio: Input channel reduction
llava/model/multimodal_encoder/mobileclip/mci.py:49
Method__init__
Construct a MobileOneBlock module. Args: in_channels: Number of channels in the input. out_channels: Number of channe
llava/model/multimodal_encoder/mobileclip/mci.py:94
Method__init__
Construct a ReparamLargeKernelConv module. Args: in_channels: Number of input channels. out_channels: Number of outpu
llava/model/multimodal_encoder/mobileclip/mci.py:377
Method__init__
(self, num_features, eps=1e-05)
llava/model/multimodal_encoder/mobileclip/mci.py:611
Method__init__
Build patch embedding layer. Args: patch_size: Patch size for embedding computation. stride: Stride for convolutional
llava/model/multimodal_encoder/mobileclip/mci.py:691
Method__init__
Build RepMixer Module. Args: dim: Input feature map dimension. :math:`C_{in}` from an expected input of size :math:`(B, C_{in}, H
llava/model/multimodal_encoder/mobileclip/mci.py:751
Method__init__
Build convolutional FFN module. Args: in_channels: Number of input channels. hidden_channels: Number of channels afte
llava/model/multimodal_encoder/mobileclip/mci.py:865
Method__init__
Build reparameterizable conditional positional encoding Args: in_channels: Number of input channels. embed_dim: Numbe
llava/model/multimodal_encoder/mobileclip/mci.py:939
Method__init__
Build RepMixer Block. Args: dim: Number of embedding dimensions. kernel_size: Kernel size for repmixer. Default: 3
llava/model/multimodal_encoder/mobileclip/mci.py:1049
Method__init__
Build Attention Block. Args: dim: Number of embedding dimensions. mlp_ratio: MLP expansion ratio. Default: 4.0
llava/model/multimodal_encoder/mobileclip/mci.py:1123
Method__init__
(self, in_dim: int, out_dim: int, *args, **kwargs)
llava/model/multimodal_encoder/mobileclip/mci.py:1275
Method__init__
( self, layers, token_mixers: Tuple[str, ...], embed_dims=None, mlp_ra
llava/model/multimodal_encoder/mobileclip/mci.py:1310
Method__init__
(self, data_path: List[str], tokenizer: transformers.PreTrainedTokenizer, da
llava/train/train.py:770
Method__init__
( self, batch_size: int, world_size: int, lengths: Optional[List[int]] = None,
llava/train/llava_trainer.py:109
Method__init__
(self, data_path: List[str], tokenizer: transformers.PreTrainedTokenizer, da
llava/train/train_qwen.py:894
Method__iter__
(self)
llava/train/llava_trainer.py:129
Method__len__
(self)
llava/train/train.py:785
Method__len__
(self)
llava/train/llava_trainer.py:126
Method__len__
(self)
llava/train/train_qwen.py:909
Method_get_train_sampler
(self)
llava/train/llava_trainer.py:139
Method_init_weights
(self, m: nn.Module)
llava/model/multimodal_encoder/mobileclip/mci.py:914
Function_prepare_decoder_attention_mask
( self, attention_mask, input_shape, inputs_embeds, past_key_values_length )
llava/train/llama_flash_attn_monkey_patch.py:98
Method_save_checkpoint
(self, model, trial, metrics=None)
llava/train/llava_trainer.py:244
Method_set_gradient_checkpointing
(self, module, value=False)
llava/model/language_model/llava_mpt.py:56
Functionadd_text
(state, text, image, image_process_mode, request: gr.Request)
llava/serve/gradio_web_server.py:128
← previousnext →101–200 of 294, ranked by callers