Views
No views yet
1import torch
2from transformers import AutoTokenizer, AutoModelForCausalLM
3
4device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
5
6id = "sirev/LFM2-2.6B-Uncensored-X64"
7
8tokenizer = AutoTokenizer.from_pretrained(id)
9
10model = AutoModelForCausalLM.from_pretrained(id).to(device)
11
12messages = [
13 {"role": "user", "content": "Your message here..."}
14]
15user = messages[0]['content']
16
17inputs = tokenizer.apply_chat_template(
18 messages,
19 add_generation_prompt=True,
20 tokenize=True,
21 return_dict=True,
22 return_tensors="pt",
23).to(device)
24
25print(f"User: {user}")
26
27outputs = model.generate(**inputs, temperature=0.3, do_sample=True, repetition_penalty=1.2, max_new_tokens=2048)
28
29print(f"AI: {tokenizer.decode(outputs[0][inputs['input_ids'].shape[-1]:], skip_special_tokens=True)}")<|startoftext|><|im_start|>system
You are a helpful assistant trained by Liquid AI.<|im_end|>
<|im_start|>user
What is C. elegans?<|im_end|>
<|im_start|>assistant
It's a tiny nematode that lives in temperate soil environments.<|im_end|>"<|im_start|>assistant" to
"<|im_start|>assistant<think>"| Benchmark (0-shot) | LFM2-2.6B-Uncensored-X64 | LiquidAI/LFM2-2.6B |
|---|---|---|
| ARC-Challenge | 45.39 % | 44.71 % |
| ARC-Easy | 58.80 % | 56.36 % |
| HellaSwag | 62.27 % | 59.71 % |
| MMLU | 63.03 % | 62.68 % |