from swarms.models import Mistral
model = Mistral(
device="cuda",
use_flash_attention=True
)
prompt = "My favourite condiment is"
result = model.run(prompt)
print(result)
