sima_utils.transformer.model.language_part_base

Classes

LanguagePartBaseModel

Base implementation for visual-language model file generation.

Module Contents

class sima_utils.transformer.model.language_part_base.LanguagePartBaseModel

Base implementation for visual-language model file generation.

cfg

Configuration of the model.

model_name

Name of the model. This will be used to determine the generated files’ names.

onnx_path

Path to store the ONNX files.

sima_path

Path to store the SiMa-specific files.

hf_model

LocalHuggingFaceModel object for obtaining the parameters to generate ONNX files.

onnx_file_name

File name of the generated ONNX file.

weight_prefix

The prefix of weight tensor names in the source model.