This model is a fine-tuned version of
meta-llama/Llama-3.1-8B-Instruct on the None dataset.
1from transformers import pipeline
2
3question = "If you had a time machine, but could only go to the past or the future once and never return, which would you choose and why?"
4generator = pipeline("text-generation", model="MaRyAm1295/Llama-3.1-8B-KAM", device="cuda")
5output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
6print(output["generated_text"])
This model was trained with SFT.