from swarms.models.mpt import MPT
mpt_instance = MPT(
"mosaicml/mpt-7b-storywriter", "EleutherAI/gpt-neox-20b", max_tokens=150
)
mpt_instance.generate("Once upon a time in a land far, far away...")
