modules/SwissArmyTransformer/sat/model/official/__init__.py (24 lines of code) (raw):

from .cuda2d_model import Cuda2dModel from .glm_model import GLMModel from .t5_model import T5Model from .vit_model import ViTModel from .bert_model import BertModel from .roberta_model import RobertaModel from .mae_model import MAE from .distill_model import DistillModel from .glm130B_model import GLM130B from .clip_model import CLIP from .gptneo_model import GPTNeoModel from .dpr_model import DPRQuestionEncoder from .dpr_model import DPRContextEncoder from .dpr_model import DPRReader from .gpt2_model import GPT2Model from .chatglm_model import ChatGLMModel from .eva2_model import EVA2Model from .llama_model import LLaMAModel from .chatglm2_model import ChatGLM2Model from .chatglm3_model import ChatGLM3Model from .chatglm4_model import ChatGLM4Model from .glm4v_model import GLM4VModel from .mixtral_model import MixtralModel from .eva_clip_model import EVA2CLIPModel