SYMBOL INDEX (112 symbols across 11 files)

FILE: setup.py
  function write_version_to_file (line 5) | def write_version_to_file(version, target_file):

FILE: struct_eqtable/__init__.py
  function get_model_name (line 15) | def get_model_name(model_path):
  function build_model (line 31) | def build_model(model_ckpt='U4R/StructTable-InternVL2-1B', **kwargs):

FILE: struct_eqtable/internvl/conversation.py
  class SeparatorStyle (line 13) | class SeparatorStyle(IntEnum):
  class Conversation (line 37) | class Conversation:
    method get_prompt (line 61) | def get_prompt(self) -> str:
    method set_system_message (line 251) | def set_system_message(self, system_message: str):
    method append_message (line 255) | def append_message(self, role: str, message: str):
    method update_last_message (line 259) | def update_last_message(self, message: str):
    method to_gradio_chatbot (line 267) | def to_gradio_chatbot(self):
    method to_openai_api_messages (line 277) | def to_openai_api_messages(self):
    method copy (line 289) | def copy(self):
    method dict (line 304) | def dict(self):
  function register_conv_template (line 318) | def register_conv_template(template: Conversation, override: bool = False):
  function get_conv_template (line 328) | def get_conv_template(name: str) -> Conversation:

FILE: struct_eqtable/internvl/internvl.py
  class InternVL (line 8) | class InternVL(nn.Module):
    method __init__ (line 9) | def __init__(self, model_path='U4R/StructTable-InternVL2-1B', max_new_...
    method init_model (line 29) | def init_model(self, model_path):
    method init_image_processor (line 39) | def init_image_processor(self, image_processor_path):
    method init_tokenizer (line 45) | def init_tokenizer(self, tokenizer_path):
    method format_image_tokens (line 58) | def format_image_tokens(self, path_num):
    method forward (line 61) | def forward(self, images, output_format='latex', **kwargs):
    method find_closest_aspect_ratio (line 130) | def find_closest_aspect_ratio(self, aspect_ratio, target_ratios, width...
    method dynamic_preprocess (line 145) | def dynamic_preprocess(self, image, min_num=1, max_num=12, image_size=...

FILE: struct_eqtable/internvl/internvl_lmdeploy.py
  class InternVL_LMDeploy (line 11) | class InternVL_LMDeploy(nn.Module):
    method __init__ (line 12) | def __init__(self, model_path='U4R/StructTable-InternVL2-1B', max_new_...
    method init_tokenizer (line 30) | def init_tokenizer(self, tokenizer_path):
    method init_model (line 37) | def init_model(self, model_path):
    method forward (line 49) | def forward(self, images, output_format='latex', **kwargs):

FILE: struct_eqtable/pix2s/pix2s.py
  class Pix2Struct (line 7) | class Pix2Struct(nn.Module):
    method __init__ (line 8) | def __init__(self, model_path='U4R/StructTable-base', max_new_tokens=1...
    method postprocess_latex_code (line 21) | def postprocess_latex_code(self, code):
    method init_model (line 26) | def init_model(self, model_path):
    method init_image_processor (line 30) | def init_image_processor(self, image_processor_path):
    method forward (line 33) | def forward(self, image, **kwargs):

FILE: struct_eqtable/pix2s/pix2s_trt.py
  function trt_dtype_to_torch (line 23) | def trt_dtype_to_torch(dtype):
  class Pix2StructTensorRT (line 36) | class Pix2StructTensorRT(nn.Module):
    method __init__ (line 38) | def __init__(self, model_path, tensorrt_path, batch_size=1, max_new_to...
    method postprocess_latex_code (line 78) | def postprocess_latex_code(self, code):
    method init_image_processor (line 83) | def init_image_processor(self):
    method init_tokenizer (line 87) | def init_tokenizer(self):
    method init_image_encoder (line 92) | def init_image_encoder(self):
    method init_llm (line 100) | def init_llm(self):
    method __call__ (line 112) | def __call__(self, image, **kwargs):
    method preprocess (line 135) | def preprocess(self, warmup, pre_prompt, post_prompt, image,
    method generate (line 168) | def generate(self, pre_prompt, post_prompt, image, decoder_input_ids,
    method get_visual_features (line 223) | def get_visual_features(self, image, attention_mask):
    method setup_fake_prompts (line 259) | def setup_fake_prompts(self, visual_features, pre_input_ids, post_inpu...
    method ptuning_setup (line 283) | def ptuning_setup(self, prompt_table, input_ids, input_lengths):
    method setup_inputs (line 312) | def setup_inputs(self, input_text, raw_image):
    method run (line 353) | def run(self, flattened_patches, attention_mask, max_new_tokens):
  function read_config (line 390) | def read_config(config_path):
  class Mapping (line 469) | class Mapping(object):
    method __init__ (line 470) | def __init__(
    method get_node_rank (line 549) | def get_node_rank(self, rank: int):
    method get_local_rank (line 552) | def get_local_rank(self, rank: int):
    method has_tp (line 555) | def has_tp(self):
    method is_last_pp_rank (line 558) | def is_last_pp_rank(self):
    method is_first_pp_rank (line 561) | def is_first_pp_rank(self):
    method has_pp (line 564) | def has_pp(self):
    method prev_pp_rank (line 567) | def prev_pp_rank(self):
    method next_pp_rank (line 573) | def next_pp_rank(self):
    method has_moe_tp (line 579) | def has_moe_tp(self):
    method has_moe_ep (line 582) | def has_moe_ep(self):
    method pp_layers (line 585) | def pp_layers(self, num_layers: int) -> List[int]:
    method ep_experts (line 591) | def ep_experts(self, num_experts: int) -> List[int]:
  function get_engine_name (line 598) | def get_engine_name(rank):
  class TRTLLMEncDecModel (line 601) | class TRTLLMEncDecModel:
    method __init__ (line 603) | def __init__(
    method from_engine (line 718) | def from_engine(cls,
    method process_input (line 734) | def process_input(self,
    method encoder_run (line 767) | def encoder_run(self,
    method generate (line 913) | def generate(self,

FILE: tools/demo/demo.py
  function parse_config (line 9) | def parse_config():
  function main (line 23) | def main():

FILE: tools/tensorrt_utils/build_visual_engine.py
  function parse_arguments (line 30) | def parse_arguments():
  class VisionEngineBuilder (line 64) | class VisionEngineBuilder:
    method __init__ (line 66) | def __init__(self, args):
    method build (line 78) | def build(self):
  function export_visual_wrapper_onnx (line 109) | def export_visual_wrapper_onnx(visual_wrapper,
  function build_trt_engine (line 127) | def build_trt_engine(model_type,
  function build_blip2_engine (line 212) | def build_blip2_engine(args):
  function build_pix2struct_engine (line 252) | def build_pix2struct_engine(args):
  function build_StructEqTable_engine (line 301) | def build_StructEqTable_engine(args):
  function build_llava_engine (line 350) | def build_llava_engine(args):
  function build_vila_engine (line 385) | def build_vila_engine(args):
  function build_nougat_engine (line 429) | def build_nougat_engine(args):
  function build_cogvlm_engine (line 457) | def build_cogvlm_engine(args):
  function build_fuyu_engine (line 493) | def build_fuyu_engine(args):
  function build_neva_engine (line 533) | def build_neva_engine(args):
  function build_video_neva_engine (line 606) | def build_video_neva_engine(args):
  function build_kosmos_engine (line 687) | def build_kosmos_engine(args):
  function build_phi_engine (line 723) | def build_phi_engine(args):

FILE: tools/tensorrt_utils/convert_checkpoint.py
  function copy_args_to_component_config (line 32) | def copy_args_to_component_config(component_config, args):
  function parse_t5_config (line 38) | def parse_t5_config(args, hf_model):
  function convert_t5_weights_to_tllm_safetensors (line 152) | def convert_t5_weights_to_tllm_safetensors(config, component, params):
  function parse_nmt_config (line 320) | def parse_nmt_config(args, model):
  function convert_nmt_weights_to_tllm_safetensors (line 441) | def convert_nmt_weights_to_tllm_safetensors(config, component, params,
  function parse_bart_config (line 598) | def parse_bart_config(args, hf_model):
  function convert_bart_weights_to_tllm_safetensors (line 748) | def convert_bart_weights_to_tllm_safetensors(config, component, params):
  function parse_pix2struct_config (line 938) | def parse_pix2struct_config(args, hf_model):
  function convert_pix2struct_weights_to_tllm_safetensors (line 1038) | def convert_pix2struct_weights_to_tllm_safetensors(config, component, pa...
  function parse_StructEqTable_config (line 1186) | def parse_StructEqTable_config(args, hf_model):
  function convert_StructEqTable_weights_to_tllm_safetensors (line 1286) | def convert_StructEqTable_weights_to_tllm_safetensors(config, component,...
  function get_model (line 1434) | def get_model(args):
  function convert_checkpoint (line 1458) | def convert_checkpoint(args):
  function convert (line 1635) | def convert(worker_rank, world_size, args, model_config, convert_args,

FILE: tools/tensorrt_utils/helper.py
  function split (line 10) | def split(v: Union[np.ndarray, torch.Tensor],
  function reshape (line 30) | def reshape(v: torch.Tensor, shape=None):
  function fuse_qkv_one_layer (line 37) | def fuse_qkv_one_layer(params, attn_module_name, trtllm_layer_name, tp_s...
  function get_qkv_module_name (line 71) | def get_qkv_module_name(model_type):
  function convert_weight_to_dtype (line 91) | def convert_weight_to_dtype(params: typing.Dict[str, torch.Tensor],