Views
No views yet
sahil2801/CodeAlpaca-20k dataset. The training data was specifically filtered to only include instructions and inputs that reference C++ or cpp, ensuring the model focuses heavily on this language domain.transformers library:1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3model_id = "VesileHan/Qwen2.5_coder_cpp"
4tokenizer = AutoTokenizer.from_pretrained(model_id)
5model = AutoModelForCausalLM.from_pretrained(
6 model_id,
7 device_map="auto",
8 torch_dtype=torch.float16
9)
10question = "How do I reverse a string in C++?"
11prompt = f"Below is an instruction that describes a coding task. Write a response that appropriately completes the request.\n\n### Instruction:\n{question}\n\n### Response:\n"
12inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
13outputs = model.generate(**inputs, max_new_tokens=150, temperature=0.7)
14print(tokenizer.decode(outputs[0][inputs['input_ids'].shape[1]:], skip_special_tokens=True))