Views
No views yet
Disclaimer: This is not an official release by Anthropic.
Claude OSS 9B is an independent open model project.


1from transformers import AutoTokenizer, AutoModelForCausalLM
2import torch
3
4model_id = "squ11z1/claude-oss-9b"
5
6tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
7model = AutoModelForCausalLM.from_pretrained(
8 model_id,
9 trust_remote_code=True,
10 torch_dtype=torch.bfloat16 if torch.cuda.is_available() else torch.float32,
11 device_map="auto",
12)
13
14messages = [{"role": "user", "content": "Who are you?"},]
15
16inputs = tokenizer.apply_chat_template(
17 messages,
18 tokenize=True,
19 add_generation_prompt=True,
20 return_tensors="pt",
21 return_dict=True,
22)
23
24inputs = {k: v.to(model.device) for k, v in inputs.items()}
25
26with torch.no_grad():
27 outputs = model.generate(
28 **inputs,
29 max_new_tokens=128,
30 do_sample=False,
31 pad_token_id=tokenizer.pad_token_id or tokenizer.eos_token_id,
32 )
33
34prompt_len = inputs["input_ids"].shape[1]
35print(tokenizer.decode(outputs[0][prompt_len:], skip_special_tokens=True))1 ./llama-cli -m claude-oss-9b-q4_k_m.gguf -p "Who are you?"
2