Views
No views yet

1pip install -r requirements.txt
2pip install transformers datasets accelerate safetensors1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3
4tokenizer = AutoTokenizer.from_pretrained("DireDreadlord/GemCod-Sapphire-270M-XM")
5model = AutoModelForCausalLM.from_pretrained("DireDreadlord/GemCod-Sapphire-270M-XM")
6model.to(device)
7model.eval()
8model.resize_token_embeddings(len(tokenizer))
9
10
11user_prompt = (
12 "write a bubble sort algorithm in cpp."
13 "Please think step by step and show your chain-of-thought before the final code." #<-- comment out this line to disable COT
14)
15
16messages = [{"role": "user", "content": user_prompt}]
17
18
19inputs = tokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt")
20inputs = {k: v.to(device) for k, v in inputs.items()}
21
22
23with torch.no_grad():
24 outputs = model.generate(
25 **inputs,
26 max_new_tokens=1024,
27 do_sample=False,
28 num_beams=1,
29 pad_token_id=tokenizer.eos_token_id,
30 eos_token_id=tokenizer.eos_token_id,
31 use_cache=False,
32 )
33
34
35prompt_len = inputs["input_ids"].shape[1]
36generated_ids = outputs[0, prompt_len:]
37print(tokenizer.decode(generated_ids.tolist(), skip_special_tokens=True))max_new_tokens=2048