test / app.py
Last commit not found
raw
history blame
235 Bytes
import transformers
from transformers import AutoModelForCausalLM
model_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
model = AutoModelForCausalLM.from_pretrained(model_id)
text = "Hello"
outputs = model.generate(text)
print(outputs)