ArmelR's picture
Update README.md
7a4a4fe
|
raw
history blame
1.29 kB
metadata
{}

Model Card for Model ID

This model is an instruction-tuned version of ⭐️ StarCoder. The instruction dataset involved is Self-instruct-starcoder which was built by boostrapping on StarCoder's generations.

Uses

Direct Use

The model was fine-tuned with the following template

Question: <instruction>

Answer: <output>

For example, your prompt can look like

instruction = "Write a function to compute the GCD between two integers a and b"
prompt = f"Question:{instruction}\n\nAnswer:"
input_ids = tokenizer(prompt, return_tensors="pt")["input_ids"]
completion = model.generate(input_ids, max_length=200)
print(tokenizer.batch_decode(completion[:,input_ids.shape[1]:])[0])