Views
No views yet
| Benchmark | 5-shot | 0-shot |
|---|---|---|
| ARC Challenge | 26.62 | 26.28 |
| ARC Easy | 51.39 | 48.11 |
| CommonsenseQA | 19.49 | 20.64 |
| HellaSWAG | 34.86 | 34.85 |
| MMLU | 27.23 | 24.87 |
| OpenBookQA | 30.20 | 30.80 |
| PIQA | 65.07 | 65.02 |
| Winogrande | 51.14 | 53.35 |
transformers library:1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
5
6model = AutoModelForCausalLM.from_pretrained("OuteAI/Lite-Oute-1-300M").to(device)
7tokenizer = AutoTokenizer.from_pretrained("OuteAI/Lite-Oute-1-300M")
8
9def generate_response(message: str, temperature: float = 0.4, repetition_penalty: float = 1.12) -> str:
10 # Convert message to PyTorch tensors
11 input_ids = tokenizer.encode(
12 message, return_tensors="pt"
13 ).to(device)
14 # Generate the response
15 output = model.generate(
16 input_ids,
17 max_length=256,
18 temperature=temperature,
19 repetition_penalty=repetition_penalty,
20 do_sample=True
21 )
22 # Decode the generated output
23 generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
24 return generated_text
25message = "Scientists have made a breakthrough in renewable energy by developing a new type of"
26response = generate_response(message)
27print(response)