Text Generation
Transformers
PyTorch
mpt
Composer
MosaicML
llm-foundry
custom_code
text-generation-inference
irenedea's picture
LLM-foundry update March 26, 2024 23:50:31
3280ec9 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