Text Generation
Transformers
PyTorch
mpt
Composer
MosaicML
llm-foundry
custom_code
text-generation-inference
mpt-7b-chat / hf.py
irenedea's picture
LLM-foundry update March 26, 2024 23:50:31
ab2b3bb verified
raw
history blame
195 Bytes
from .hf_causal_lm import ComposerHFCausalLM
from .hf_fsdp import prepare_hf_causal_lm_model_for_fsdp, prepare_hf_enc_dec_model_for_fsdp, prepare_hf_model_for_fsdp
from .hf_t5 import ComposerHFT5