Views
No views yet
Question: <instruction>
Answer: <output>1instruction = "Write a function to compute the GCD between two integers a and b"
2prompt = f"Question:{instruction}\n\nAnswer:"
3input_ids = tokenizer(prompt, return_tensors="pt")["input_ids"]
4completion = model.generate(input_ids, max_length=200)
5print(tokenizer.batch_decode(completion[:,input_ids.shape[1]:])[0])