SYMBOL INDEX (335 symbols across 32 files)

FILE: llava/conversation.py
  class SeparatorStyle (line 9) | class SeparatorStyle(Enum):
  class Conversation (line 21) | class Conversation:
    method get_prompt (line 34) | def get_prompt(self):
    method append_message (line 143) | def append_message(self, role, message):
    method process_image (line 146) | def process_image(self, image, image_process_mode, return_pil=False, i...
    method get_images (line 186) | def get_images(self, return_pil=False):
    method to_gradio_chatbot (line 196) | def to_gradio_chatbot(self):
    method copy (line 214) | def copy(self):
    method dict (line 225) | def dict(self):

FILE: llava/mm_utils.py
  function select_best_resolution (line 14) | def select_best_resolution(original_size, possible_resolutions):
  function resize_and_pad_image (line 44) | def resize_and_pad_image(image, target_resolution):
  function divide_to_patches (line 79) | def divide_to_patches(image, patch_size):
  function get_anyres_image_grid_shape (line 101) | def get_anyres_image_grid_shape(image_size, grid_pinpoints, patch_size):
  function process_anyres_image (line 121) | def process_anyres_image(image, processor, grid_pinpoints):
  function load_image_from_base64 (line 150) | def load_image_from_base64(image):
  function expand2square (line 154) | def expand2square(pil_img, background_color):
  function process_images (line 168) | def process_images(images, image_processor, model_cfg):
  function tokenizer_image_token (line 187) | def tokenizer_image_token(prompt, tokenizer, image_token_index=IMAGE_TOK...
  function get_model_name_from_path (line 209) | def get_model_name_from_path(model_path):
  class KeywordsStoppingCriteria (line 218) | class KeywordsStoppingCriteria(StoppingCriteria):
    method __init__ (line 219) | def __init__(self, keywords, tokenizer, input_ids):
    method call_for_batch (line 233) | def call_for_batch(self, output_ids: torch.LongTensor, scores: torch.F...
    method __call__ (line 246) | def __call__(self, output_ids: torch.LongTensor, scores: torch.FloatTe...

FILE: llava/model/apply_delta.py
  function apply_delta (line 13) | def apply_delta(base_model_path, target_model_path, delta_path):

FILE: llava/model/builder.py
  function load_pretrained_model (line 26) | def load_pretrained_model(model_path, model_base, model_name, load_8bit=...

FILE: llava/model/consolidate.py
  function consolidate_ckpt (line 13) | def consolidate_ckpt(src_path, dst_path):

FILE: llava/model/language_model/llava_llama.py
  class LlavaConfig (line 30) | class LlavaConfig(LlamaConfig):
  class LlavaLlamaModel (line 34) | class LlavaLlamaModel(LlavaMetaModel, LlamaModel):
    method __init__ (line 37) | def __init__(self, config: LlamaConfig):
  class LlavaLlamaForCausalLM (line 41) | class LlavaLlamaForCausalLM(LlamaForCausalLM, LlavaMetaForCausalLM):
    method __init__ (line 44) | def __init__(self, config):
    method get_model (line 54) | def get_model(self):
    method forward (line 57) | def forward(
    method generate (line 106) | def generate(
    method prepare_inputs_for_generation (line 145) | def prepare_inputs_for_generation(self, input_ids, past_key_values=None,

FILE: llava/model/language_model/llava_mistral.py
  class LlavaMistralConfig (line 31) | class LlavaMistralConfig(MistralConfig):
  class LlavaMistralModel (line 35) | class LlavaMistralModel(LlavaMetaModel, MistralModel):
    method __init__ (line 38) | def __init__(self, config: MistralConfig):
  class LlavaMistralForCausalLM (line 42) | class LlavaMistralForCausalLM(MistralForCausalLM, LlavaMetaForCausalLM):
    method __init__ (line 45) | def __init__(self, config):
    method get_model (line 54) | def get_model(self):
    method forward (line 57) | def forward(
    method generate (line 105) | def generate(
    method prepare_inputs_for_generation (line 144) | def prepare_inputs_for_generation(self, input_ids, past_key_values=None,

FILE: llava/model/language_model/llava_mpt.py
  class LlavaMptConfig (line 25) | class LlavaMptConfig(MptConfig):
  class LlavaMptModel (line 29) | class LlavaMptModel(LlavaMetaModel, MptModel):
    method __init__ (line 32) | def __init__(self, config: MptConfig):
    method embed_tokens (line 36) | def embed_tokens(self, x):
  class LlavaMptForCausalLM (line 40) | class LlavaMptForCausalLM(MptForCausalLM, LlavaMetaForCausalLM):
    method __init__ (line 44) | def __init__(self, config):
    method get_model (line 53) | def get_model(self):
    method _set_gradient_checkpointing (line 56) | def _set_gradient_checkpointing(self, module, value=False):
    method forward (line 60) | def forward(
    method prepare_inputs_for_generation (line 87) | def prepare_inputs_for_generation(self, input_ids, past_key_values=Non...

FILE: llava/model/language_model/llava_qwen.py
  class LlavaConfig (line 30) | class LlavaConfig(Qwen2Config):
  class LlavaQwen2Model (line 34) | class LlavaQwen2Model(LlavaMetaModel, Qwen2Model):
    method __init__ (line 37) | def __init__(self, config: Qwen2Config):
  class LlavaQwen2ForCausalLM (line 41) | class LlavaQwen2ForCausalLM(Qwen2ForCausalLM, LlavaMetaForCausalLM):
    method __init__ (line 44) | def __init__(self, config):
    method get_model (line 54) | def get_model(self):
    method forward (line 57) | def forward(
    method generate (line 106) | def generate(
    method prepare_inputs_for_generation (line 145) | def prepare_inputs_for_generation(self, input_ids, past_key_values=None,

FILE: llava/model/llava_arch.py
  class LlavaMetaModel (line 29) | class LlavaMetaModel:
    method __init__ (line 31) | def __init__(self, config):
    method get_vision_tower (line 43) | def get_vision_tower(self):
    method initialize_vision_modules (line 49) | def initialize_vision_modules(self, model_args, fsdp=None):
  function unpad_image (line 101) | def unpad_image(tensor, original_size):
  class LlavaMetaForCausalLM (line 132) | class LlavaMetaForCausalLM(ABC):
    method get_model (line 135) | def get_model(self):
    method get_vision_tower (line 138) | def get_vision_tower(self):
    method encode_images (line 141) | def encode_images(self, images):
    method prepare_inputs_labels_for_multimodal (line 146) | def prepare_inputs_labels_for_multimodal(
    method initialize_vision_tokenizer (line 334) | def initialize_vision_tokenizer(self, model_args, tokenizer):

FILE: llava/model/make_delta.py
  function make_delta (line 13) | def make_delta(base_model_path, target_model_path, delta_path, hub_repo_...

FILE: llava/model/multimodal_encoder/builder.py
  function build_vision_tower (line 6) | def build_vision_tower(vision_tower_cfg, **kwargs):

FILE: llava/model/multimodal_encoder/clip_encoder.py
  class CLIPVisionTower (line 7) | class CLIPVisionTower(nn.Module):
    method __init__ (line 8) | def __init__(self, vision_tower, args, delay_load=False):
    method load_model (line 31) | def load_model(self, device_map=None):
    method feature_select (line 48) | def feature_select(self, image_forward_outs):
    method forward (line 58) | def forward(self, images):
    method forward_images (line 65) | def forward_images(self, images):
    method dummy_feature (line 79) | def dummy_feature(self):
    method dtype (line 83) | def dtype(self):
    method device (line 87) | def device(self):
    method config (line 91) | def config(self):
    method hidden_size (line 98) | def hidden_size(self):
    method num_patches_per_side (line 102) | def num_patches_per_side(self):
    method num_patches (line 106) | def num_patches(self):
  class CLIPVisionTowerS2 (line 111) | class CLIPVisionTowerS2(CLIPVisionTower):
    method __init__ (line 112) | def __init__(self, vision_tower, args, delay_load=False):
    method load_model (line 132) | def load_model(self, device_map=None):
    method forward_feature (line 147) | def forward_feature(self, images):
    method forward (line 153) | def forward(self, images):
    method hidden_size (line 165) | def hidden_size(self):

FILE: llava/model/multimodal_encoder/mobileclip/__init__.py
  function load_model_config (line 15) | def load_model_config(
  class MCi (line 34) | class MCi(nn.Module):
    method __init__ (line 39) | def __init__(self, model_name: str, *args, **kwargs) -> None:
    method forward (line 55) | def forward(self, x: Any, *args, **kwargs) -> Any:
    method _get_in_feature_dimension (line 61) | def _get_in_feature_dimension(image_classifier: nn.Module) -> int:
    method _update_image_classifier (line 82) | def _update_image_classifier(

FILE: llava/model/multimodal_encoder/mobileclip/mci.py
  function _cfg (line 20) | def _cfg(url="", **kwargs):
  class SEBlock (line 42) | class SEBlock(nn.Module):
    method __init__ (line 49) | def __init__(self, in_channels: int, rd_ratio: float = 0.0625) -> None:
    method forward (line 72) | def forward(self, inputs: torch.Tensor) -> torch.Tensor:
  class MobileOneBlock (line 84) | class MobileOneBlock(nn.Module):
    method __init__ (line 94) | def __init__(
    method forward (line 194) | def forward(self, x: torch.Tensor) -> torch.Tensor:
    method reparameterize (line 219) | def reparameterize(self):
    method _get_kernel_bias (line 249) | def _get_kernel_bias(self) -> Tuple[torch.Tensor, torch.Tensor]:
    method _fuse_bn_tensor (line 284) | def _fuse_bn_tensor(
    method _conv_bn (line 332) | def _conv_bn(self, kernel_size: int, padding: int) -> nn.Sequential:
  class ReparamLargeKernelConv (line 368) | class ReparamLargeKernelConv(nn.Module):
    method __init__ (line 377) | def __init__(
    method forward (line 442) | def forward(self, x: torch.Tensor) -> torch.Tensor:
    method get_kernel_bias (line 453) | def get_kernel_bias(self) -> Tuple[torch.Tensor, torch.Tensor]:
    method reparameterize (line 469) | def reparameterize(self) -> None:
    method _fuse_bn (line 495) | def _fuse_bn(
    method _conv_bn (line 517) | def _conv_bn(self, kernel_size: int, padding: int = 0) -> nn.Sequential:
  function convolutional_stem (line 553) | def convolutional_stem(
  class LayerNormChannel (line 606) | class LayerNormChannel(nn.Module):
    method __init__ (line 611) | def __init__(self, num_features, eps=1e-05) -> None:
    method forward (line 617) | def forward(self, x) -> torch.Tensor:
  class MHSA (line 626) | class MHSA(nn.Module):
    method __init__ (line 633) | def __init__(
    method forward (line 661) | def forward(self, x: torch.Tensor) -> torch.Tensor:
  class PatchEmbed (line 688) | class PatchEmbed(nn.Module):
    method __init__ (line 691) | def __init__(
    method forward (line 739) | def forward(self, x: torch.Tensor) -> torch.Tensor:
  class RepMixer (line 744) | class RepMixer(nn.Module):
    method __init__ (line 751) | def __init__(
    method forward (line 808) | def forward(self, x: torch.Tensor) -> torch.Tensor:
    method reparameterize (line 819) | def reparameterize(self) -> None:
  class ConvFFN (line 862) | class ConvFFN(nn.Module):
    method __init__ (line 865) | def __init__(
    method _init_weights (line 914) | def _init_weights(self, m: nn.Module) -> None:
    method forward (line 920) | def forward(self, x: torch.Tensor) -> torch.Tensor:
  class RepCPE (line 930) | class RepCPE(nn.Module):
    method __init__ (line 939) | def __init__(
    method forward (line 992) | def forward(self, x: torch.Tensor) -> torch.Tensor:
    method reparameterize (line 1000) | def reparameterize(self) -> None:
  class RepMixerBlock (line 1042) | class RepMixerBlock(nn.Module):
    method __init__ (line 1049) | def __init__(
    method forward (line 1106) | def forward(self, x):
  class AttentionBlock (line 1116) | class AttentionBlock(nn.Module):
    method __init__ (line 1123) | def __init__(
    method forward (line 1185) | def forward(self, x):
  function basic_blocks (line 1195) | def basic_blocks(
  class GlobalPool2D (line 1272) | class GlobalPool2D(nn.Module):
    method __init__ (line 1275) | def __init__(self, in_dim: int, out_dim: int, *args, **kwargs) -> None:
    method pool (line 1282) | def pool(self, x) -> Tensor:
    method forward (line 1290) | def forward(self, x: Tensor, *args, **kwargs) -> Tensor:
  class FastViT (line 1305) | class FastViT(nn.Module):
    method __init__ (line 1310) | def __init__(
    method cls_init_weights (line 1420) | def cls_init_weights(self, m: nn.Module) -> None:
    method forward_embeddings (line 1427) | def forward_embeddings(self, x: torch.Tensor) -> torch.Tensor:
    method forward_tokens (line 1431) | def forward_tokens(self, x: torch.Tensor, *args, **kwargs) -> torch.Te...
    method forward (line 1436) | def forward(self, x: torch.Tensor, *args, **kwargs) -> Union[Tensor, D...
  function fastvithd (line 1455) | def fastvithd(pretrained=False, **kwargs):

FILE: llava/model/multimodal_encoder/mobileclip_encoder.py
  class MobileCLIPVisionTower (line 13) | class MobileCLIPVisionTower(nn.Module):
    method __init__ (line 14) | def __init__(self, vision_tower, args, delay_load=False):
    method load_model (line 31) | def load_model(self, device_map=None):
    method feature_select (line 60) | def feature_select(self, image_forward_outs):
    method forward (line 70) | def forward(self, images):
    method forward_images (line 77) | def forward_images(self, images):
    method dummy_feature (line 91) | def dummy_feature(self):
    method dtype (line 95) | def dtype(self):
    method device (line 99) | def device(self):
    method config (line 103) | def config(self):
    method hidden_size (line 107) | def hidden_size(self):
    method num_patches_per_side (line 111) | def num_patches_per_side(self):
    method num_patches (line 115) | def num_patches(self):

FILE: llava/model/multimodal_projector/builder.py
  class IdentityMap (line 5) | class IdentityMap(nn.Module):
    method __init__ (line 6) | def __init__(self):
    method forward (line 9) | def forward(self, x, *args, **kwargs):
    method config (line 13) | def config(self):
  function build_vision_projector (line 17) | def build_vision_projector(config, delay_load=False, **kwargs):

FILE: llava/model/utils.py
  function auto_upgrade (line 4) | def auto_upgrade(config):

FILE: llava/serve/cli.py
  function load_image (line 18) | def load_image(image_file):
  function main (line 27) | def main(args):

FILE: llava/serve/controller.py
  class DispatchMethod (line 28) | class DispatchMethod(Enum):
    method from_str (line 33) | def from_str(cls, name):
  class WorkerInfo (line 43) | class WorkerInfo:
  function heart_beat_controller (line 51) | def heart_beat_controller(controller):
  class Controller (line 57) | class Controller:
    method __init__ (line 58) | def __init__(self, dispatch_method: str):
    method register_worker (line 69) | def register_worker(self, worker_name: str, check_heart_beat: bool,
    method get_worker_status (line 88) | def get_worker_status(self, worker_name: str):
    method remove_worker (line 101) | def remove_worker(self, worker_name: str):
    method refresh_all_workers (line 104) | def refresh_all_workers(self):
    method list_models (line 112) | def list_models(self):
    method get_worker_address (line 120) | def get_worker_address(self, model_name: str):
    method receive_heart_beat (line 173) | def receive_heart_beat(self, worker_name: str, queue_length: int):
    method remove_stable_workers_by_expiration (line 183) | def remove_stable_workers_by_expiration(self):
    method worker_api_generate_stream (line 193) | def worker_api_generate_stream(self, params):
    method worker_api_get_status (line 220) | def worker_api_get_status(self):
  function register_worker (line 243) | async def register_worker(request: Request):
  function refresh_all_workers (line 251) | async def refresh_all_workers():
  function list_models (line 256) | async def list_models():
  function get_worker_address (line 262) | async def get_worker_address(request: Request):
  function receive_heart_beat (line 269) | async def receive_heart_beat(request: Request):
  function worker_api_generate_stream (line 277) | async def worker_api_generate_stream(request: Request):
  function worker_api_get_status (line 284) | async def worker_api_get_status(request: Request):

FILE: llava/serve/gradio_web_server.py
  function get_conv_log_filename (line 32) | def get_conv_log_filename():
  function get_model_list (line 38) | def get_model_list():
  function load_demo (line 58) | def load_demo(url_params, request: gr.Request):
  function load_demo_refresh_model_list (line 71) | def load_demo_refresh_model_list(request: gr.Request):
  function vote_last_response (line 82) | def vote_last_response(state, vote_type, model_selector, request: gr.Req...
  function upvote_last_response (line 94) | def upvote_last_response(state, model_selector, request: gr.Request):
  function downvote_last_response (line 100) | def downvote_last_response(state, model_selector, request: gr.Request):
  function flag_last_response (line 106) | def flag_last_response(state, model_selector, request: gr.Request):
  function regenerate (line 112) | def regenerate(state, image_process_mode, request: gr.Request):
  function clear_history (line 122) | def clear_history(request: gr.Request):
  function add_text (line 128) | def add_text(state, text, image, image_process_mode, request: gr.Request):
  function http_bot (line 154) | def http_bot(state, model_selector, temperature, top_p, max_new_tokens, ...
  function build_demo (line 317) | def build_demo(embed_mode, cur_dir=None, concurrency_count=10):

FILE: llava/serve/model_worker.py
  function heart_beat_worker (line 37) | def heart_beat_worker(controller):
  class ModelWorker (line 44) | class ModelWorker:
    method __init__ (line 45) | def __init__(self, controller_addr, worker_addr,
    method register_to_controller (line 75) | def register_to_controller(self):
    method send_heart_beat (line 87) | def send_heart_beat(self):
    method get_queue_length (line 108) | def get_queue_length(self):
    method get_status (line 115) | def get_status(self):
    method generate_stream (line 123) | def generate_stream(self, params):
    method generate_stream_gate (line 195) | def generate_stream_gate(self, params):
  function release_model_semaphore (line 225) | def release_model_semaphore(fn=None):
  function generate_stream (line 232) | async def generate_stream(request: Request):
  function get_status (line 248) | async def get_status(request: Request):

FILE: llava/serve/sglang_worker.py
  function heart_beat_worker (line 38) | def heart_beat_worker(controller):
  function pipeline (line 45) | def pipeline(s, prompt, max_tokens):
  class ModelWorker (line 54) | class ModelWorker:
    method __init__ (line 55) | def __init__(self, controller_addr, worker_addr, sgl_endpoint,
    method register_to_controller (line 85) | def register_to_controller(self):
    method send_heart_beat (line 97) | def send_heart_beat(self):
    method get_queue_length (line 118) | def get_queue_length(self):
    method get_status (line 125) | def get_status(self):
    method generate_stream (line 132) | async def generate_stream(self, params):
    method generate_stream_gate (line 172) | async def generate_stream_gate(self, params):
  function release_model_semaphore (line 195) | def release_model_semaphore(fn=None):
  function generate_stream (line 202) | async def generate_stream(request: Request):
  function get_status (line 218) | async def get_status(request: Request):

FILE: llava/serve/test_message.py
  function main (line 9) | def main():

FILE: llava/train/llama_flash_attn_monkey_patch.py
  function forward (line 16) | def forward(
  function _prepare_decoder_attention_mask (line 98) | def _prepare_decoder_attention_mask(
  function replace_llama_attn_with_flash_attn (line 105) | def replace_llama_attn_with_flash_attn():

FILE: llava/train/llama_xformers_attn_monkey_patch.py
  function replace_llama_attn_with_xformers_attn (line 19) | def replace_llama_attn_with_xformers_attn():
  function xformers_forward (line 23) | def xformers_forward(

FILE: llava/train/llava_trainer.py
  function maybe_zero_3 (line 22) | def maybe_zero_3(param, ignore_status=False, name=None):
  function get_mm_adapter_state_maybe_zero_3 (line 36) | def get_mm_adapter_state_maybe_zero_3(named_params, keys_to_match):
  function split_to_even_chunks (line 42) | def split_to_even_chunks(indices, lengths, num_chunks):
  function get_modality_length_grouped_indices (line 64) | def get_modality_length_grouped_indices(lengths, batch_size, world_size,...
  function get_length_grouped_indices (line 92) | def get_length_grouped_indices(lengths, batch_size, world_size, generato...
  class LengthGroupedSampler (line 103) | class LengthGroupedSampler(Sampler):
    method __init__ (line 109) | def __init__(
    method __len__ (line 126) | def __len__(self):
    method __iter__ (line 129) | def __iter__(self):
  class LLaVATrainer (line 137) | class LLaVATrainer(Trainer):
    method _get_train_sampler (line 139) | def _get_train_sampler(self) -> Optional[torch.utils.data.Sampler]:
    method create_optimizer (line 154) | def create_optimizer(self):
    method _save_checkpoint (line 244) | def _save_checkpoint(self, model, trial, metrics=None):
    method _save (line 267) | def _save(self, output_dir: Optional[str] = None, state_dict=None):

FILE: llava/train/train.py
  function rank0_print (line 45) | def rank0_print(*args):
  class ModelArguments (line 54) | class ModelArguments:
  class DataArguments (line 74) | class DataArguments:
  class TrainingArguments (line 87) | class TrainingArguments(transformers.TrainingArguments):
  function maybe_zero_3 (line 123) | def maybe_zero_3(param, ignore_status=False, name=None):
  function get_peft_state_maybe_zero_3 (line 138) | def get_peft_state_maybe_zero_3(named_params, bias):
  function get_peft_state_non_lora_maybe_zero_3 (line 163) | def get_peft_state_non_lora_maybe_zero_3(named_params, require_grad_only...
  function get_mm_adapter_state_maybe_zero_3 (line 171) | def get_mm_adapter_state_maybe_zero_3(named_params, keys_to_match):
  function find_all_linear_names (line 177) | def find_all_linear_names(model):
  function safe_save_model_for_hf_trainer (line 193) | def safe_save_model_for_hf_trainer(trainer: transformers.Trainer,
  function smart_tokenizer_and_embedding_resize (line 232) | def smart_tokenizer_and_embedding_resize(
  function _tokenize_fn (line 257) | def _tokenize_fn(strings: Sequence[str],
  function _mask_targets (line 284) | def _mask_targets(target, tokenized_lens, speakers):
  function _add_speaker_and_signal (line 295) | def _add_speaker_and_signal(header, source, get_conversation=True):
  function preprocess_multimodal (line 316) | def preprocess_multimodal(
  function preprocess_llama_2 (line 340) | def preprocess_llama_2(
  function preprocess_qwen_2 (line 423) | def preprocess_qwen_2(
  function preprocess_v1 (line 519) | def preprocess_v1(
  function preprocess_mpt (line 605) | def preprocess_mpt(
  function preprocess_plain (line 693) | def preprocess_plain(
  function preprocess (line 715) | def preprocess(
  class LazySupervisedDataset (line 767) | class LazySupervisedDataset(Dataset):
    method __init__ (line 770) | def __init__(self, data_path: List[str],
    method __len__ (line 785) | def __len__(self):
    method lengths (line 789) | def lengths(self):
    method modality_lengths (line 797) | def modality_lengths(self):
    method __getitem__ (line 805) | def __getitem__(self, i) -> Dict[str, torch.Tensor]:
  class DataCollatorForSupervisedDataset (line 863) | class DataCollatorForSupervisedDataset(object):
    method __call__ (line 868) | def __call__(self, instances: Sequence[Dict]) -> Dict[str, torch.Tensor]:
  function make_supervised_data_module (line 897) | def make_supervised_data_module(tokenizer: transformers.PreTrainedTokeni...
  function train (line 909) | def train(attn_implementation=None):

FILE: llava/train/train_qwen.py
  function rank0_print (line 45) | def rank0_print(*args):
  class ModelArguments (line 54) | class ModelArguments:
  class DataArguments (line 74) | class DataArguments:
  class TrainingArguments (line 87) | class TrainingArguments(transformers.TrainingArguments):
  function maybe_zero_3 (line 123) | def maybe_zero_3(param, ignore_status=False, name=None):
  function get_peft_state_maybe_zero_3 (line 138) | def get_peft_state_maybe_zero_3(named_params, bias):
  function get_peft_state_non_lora_maybe_zero_3 (line 163) | def get_peft_state_non_lora_maybe_zero_3(named_params, require_grad_only...
  function get_mm_adapter_state_maybe_zero_3 (line 171) | def get_mm_adapter_state_maybe_zero_3(named_params, keys_to_match):
  function find_all_linear_names (line 177) | def find_all_linear_names(model):
  function safe_save_model_for_hf_trainer (line 193) | def safe_save_model_for_hf_trainer(trainer: transformers.Trainer,
  function smart_tokenizer_and_embedding_resize (line 232) | def smart_tokenizer_and_embedding_resize(
  function _tokenize_fn (line 257) | def _tokenize_fn(strings: Sequence[str],
  function _mask_targets (line 284) | def _mask_targets(target, tokenized_lens, speakers):
  function _add_speaker_and_signal (line 295) | def _add_speaker_and_signal(header, source, get_conversation=True):
  function preprocess_multimodal (line 316) | def preprocess_multimodal(
  function preprocess_llama_2 (line 340) | def preprocess_llama_2(
  function preprocess_qwen_2 (line 527) | def preprocess_qwen_2(
  function preprocess_v1 (line 637) | def preprocess_v1(
  function preprocess_mpt (line 723) | def preprocess_mpt(
  function preprocess_plain (line 811) | def preprocess_plain(
  function preprocess (line 833) | def preprocess(
  class LazySupervisedDataset (line 891) | class LazySupervisedDataset(Dataset):
    method __init__ (line 894) | def __init__(self, data_path: List[str],
    method __len__ (line 909) | def __len__(self):
    method lengths (line 913) | def lengths(self):
    method modality_lengths (line 921) | def modality_lengths(self):
    method get_sample (line 929) | def get_sample(self, i) -> Dict[str, torch.Tensor]:
    method __getitem__ (line 985) | def __getitem__(self, i) -> Dict[str, torch.Tensor]:
  class DataCollatorForSupervisedDataset (line 995) | class DataCollatorForSupervisedDataset(object):
    method __call__ (line 1000) | def __call__(self, instances: Sequence[Dict]) -> Dict[str, torch.Tensor]:
  function make_supervised_data_module (line 1029) | def make_supervised_data_module(tokenizer: transformers.PreTrainedTokeni...
  function train (line 1041) | def train(attn_implementation=None):

FILE: llava/utils.py
  function build_logger (line 17) | def build_logger(logger_name, logger_filename):
  class StreamToLogger (line 60) | class StreamToLogger(object):
    method __init__ (line 65) | def __init__(self, logger, log_level=logging.INFO):
    method __getattr__ (line 71) | def __getattr__(self, attr):
    method write (line 74) | def write(self, buf):
    method flush (line 88) | def flush(self):
  function disable_torch_init (line 94) | def disable_torch_init():
  function violates_moderation (line 103) | def violates_moderation(text):
  function pretty_print_semaphore (line 124) | def pretty_print_semaphore(semaphore):

FILE: model_export/export_vision_encoder.py
  function export (line 19) | def export(args):

FILE: predict.py
  function predict (line 18) | def predict(args):