from transformers import pipeline

question = "Write a Python function to implement insertion sort on a list of floating-point numbers and return the sorted list."
generator = pipeline("text-generation", model="Luca0867/selfinstruct")
output = generator([{"role": "user", "content": question}], max_new_tokens=1024, return_full_text=False)[0]
print(output["generated_text"])

Training procedure

This model was trained with SFT.

Framework versions

TRL: 1.1.0
Transformers: 5.5.4
Pytorch: 2.10.0+cu128
Datasets: 4.8.4
Tokenizers: 0.22.2