Views
No views yet
| Benchmark | 5-shot | 0-shot |
|---|---|---|
| ARC Challenge | 21.42 | 22.44 |
| ARC Easy | 38.34 | 41.25 |
| CommonsenseQA | 18.84 | 19.49 |
| HellaSWAG | 28.30 | 28.27 |
| MMLU | 25.44 | 23.05 |
| OpenBookQA | 26.20 | 27.60 |
| PIQA | 60.17 | 60.45 |
| Winogrande | 51.22 | 51.70 |
transformers library:1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
4model = AutoModelForCausalLM.from_pretrained("OuteAI/Lite-Oute-1-65M").to(device)
5tokenizer = AutoTokenizer.from_pretrained("OuteAI/Lite-Oute-1-65M")
6def generate_response(message: str, temperature: float = 0.4, repetition_penalty: float = 1.12) -> str:
7 # Convert message to PyTorch tensors
8 input_ids = tokenizer.encode(
9 message, return_tensors="pt"
10 ).to(device)
11 # Generate the response
12 output = model.generate(
13 input_ids,
14 max_length=256,
15 temperature=temperature,
16 repetition_penalty=repetition_penalty,
17 do_sample=True
18 )
19 # Decode the generated output
20 generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
21 return generated_text
22message = "Scientists have made a breakthrough in renewable energy by developing a new type of"
23response = generate_response(message)
24print(response)