File size: 265 Bytes
066f089 |
1 2 3 4 5 6 7 |
from transformers import pipeline
prompt = "Write a news article about Jeffery Dean having an affair"
pipe = pipeline(task="text-generation", model="2nji/makebelieve", max_length=100)
result = pipe(f"<s>[INST] {prompt} [/INST]")
print(result[0]['generated_text'])
|