You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
10 lines
181 B
10 lines
181 B
1 year ago
|
from swarms.models import Mistral
|
||
|
|
|
||
|
|
model = Mistral(
|
||
|
|
device="cuda",
|
||
|
|
use_flash_attention=True
|
||
|
|
)
|
||
|
|
|
||
|
|
prompt = "My favourite condiment is"
|
||
|
|
result = model.run(prompt)
|
||
|
|
print(result)
|