from swarms.models import Mistral

model = Mistral(device="cuda", use_flash_attention=True)

prompt = "My favourite condiment is"
result = model.run(prompt)
print(result)