File size: 695 Bytes
de4ade4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
# Copyright 2022 MosaicML LLM Foundry authors
# SPDX-License-Identifier: Apache-2.0
from llmfoundry.models.hf import (ComposerHFCausalLM, ComposerHFPrefixLM,
ComposerHFT5)
from llmfoundry.models.inference_api_wrapper import (OpenAICausalLMEvalWrapper,
OpenAIChatAPIEvalWrapper)
from llmfoundry.models.mpt import ComposerMPTCausalLM
COMPOSER_MODEL_REGISTRY = {
'mpt_causal_lm': ComposerMPTCausalLM,
'hf_causal_lm': ComposerHFCausalLM,
'hf_prefix_lm': ComposerHFPrefixLM,
'hf_t5': ComposerHFT5,
'openai_causal_lm': OpenAICausalLMEvalWrapper,
'openai_chat': OpenAIChatAPIEvalWrapper
}
|