Views
No views yet
This repository stores an unchanged copy ofstabilityai/stable-code-instruct-3b
for use as a base model in future fine‑tuning projects (including Stephen).
stable-code-instruct-3b is a 2.7B parameter decoder-only transformer from Stability AI, tuned for multi‑language code generation and conversational coding assistance.| Language | pass@1 |
|---|---|
| Python | 32.4% |
| C++ | 30.9% |
| Java | 32.1% |
| JavaScript | 32.1% |
| PHP | 24.2% |
| Rust | 23.0% |
1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4model_id = "dgtalbug/stable-code-instruct-3b"
5tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
6model = AutoModelForCausalLM.from_pretrained(
7 model_id, torch_dtype=torch.bfloat16, trust_remote_code=True
8).cuda().eval()
9
10messages = [
11 {"role": "system", "content": "You are a helpful coding assistant."},
12 {"role": "user", "content": "Write a Python function to reverse a string."}
13]
14
15prompt = tokenizer.apply_chat_template(messages, add_generation_prompt=True, tokenize=False)
16inputs = tokenizer([prompt], return_tensors="pt").to(model.device)
17
18tokens = model.generate(
19 **inputs,
20 max_new_tokens=200,
21 temperature=0.5,
22 top_p=0.95,
23 top_k=100,
24 do_sample=True,
25 use_cache=True
26)
27
28output = tokenizer.batch_decode(tokens[:, inputs.input_ids.shape[-1]:], skip_special_tokens=True)[0]
29print(output)