Views
No views yet

1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_id = "macadeliccc/CapyLake-7B-v2-laser"
4tokenizer = AutoTokenizer.from_pretrained(model_id)
5model = AutoModelForCausalLM.from_pretrained(model_id)
6
7text = "Create an idea for a TV show and write a short pilot script"
8inputs = tokenizer(text, return_tensors="pt")
9
10# Adding hyperparameters to the generation call
11outputs = model.generate(
12 **inputs,
13 max_new_tokens=4096, # Controls the maximum length of the new tokens created
14 temperature=0.7, # Adjust for creativity (lower is less random)
15 top_k=50, # Keeps the top k tokens for sampling
16 top_p=0.95, # Uses nucleus sampling with this cumulative probability
17 num_return_sequences=1, # Number of sequences to generate
18 no_repeat_ngram_size=2, # Prevents repeating n-grams to ensure diversity
19 early_stopping=True # Stops generation when all sequences reach the EOS token
20)
21
22print(tokenizer.decode(outputs[0], skip_special_tokens=True))| Model | AGIEval | GPT4All | TruthfulQA | Bigbench | Average |
|---|---|---|---|---|---|
| CapyLake-7B-v2-laser | 44.34 | 77.77 | 68.47 | 47.92 | 59.62 |
| Task | Version | Metric | Value | Stderr | |
|---|---|---|---|---|---|
| agieval_aqua_rat | 0 | acc | 28.35 | ± | 2.83 |
| acc_norm | 25.98 | ± | 2.76 | ||
| agieval_logiqa_en | 0 | acc | 38.86 | ± | 1.91 |
| acc_norm | 39.02 | ± | 1.91 | ||
| agieval_lsat_ar | 0 | acc | 25.22 | ± | 2.87 |
| acc_norm | 24.35 | ± | 2.84 | ||
| agieval_lsat_lr | 0 | acc | 50.39 | ± | 2.22 |
| acc_norm | 51.57 | ± | 2.22 | ||
| agieval_lsat_rc | 0 | acc | 65.06 | ± | 2.91 |
| acc_norm | 63.94 | ± | 2.93 | ||
| agieval_sat_en | 0 | acc | 78.64 | ± | 2.86 |
| acc_norm | 78.64 | ± | 2.86 | ||
| agieval_sat_en_without_passage | 0 | acc | 40.78 | ± | 3.43 |
| acc_norm | 40.78 | ± | 3.43 | ||
| agieval_sat_math | 0 | acc | 33.64 | ± | 3.19 |
| acc_norm | 30.45 | ± | 3.11 |
| Task | Version | Metric | Value | Stderr | |
|---|---|---|---|---|---|
| arc_challenge | 0 | acc | 66.89 | ± | 1.38 |
| acc_norm | 67.49 | ± | 1.37 | ||
| arc_easy | 0 | acc | 86.70 | ± | 0.70 |
| acc_norm | 81.90 | ± | 0.79 | ||
| boolq | 1 | acc | 88.10 | ± | 0.57 |
| hellaswag | 0 | acc | 71.45 | ± | 0.45 |
| acc_norm | 87.78 | ± | 0.33 | ||
| openbookqa | 0 | acc | 39.80 | ± | 2.19 |
| acc_norm | 49.80 | ± | 2.24 | ||
| piqa | 0 | acc | 82.86 | ± | 0.88 |
| acc_norm | 84.87 | ± | 0.84 | ||
| winogrande | 0 | acc | 84.45 | ± | 1.02 |
| Task | Version | Metric | Value | Stderr | |
|---|---|---|---|---|---|
| truthfulqa_mc | 1 | mc1 | 53.98 | ± | 1.74 |
| mc2 | 68.47 | ± | 1.53 |
| Task | Version | Metric | Value | Stderr | |
|---|---|---|---|---|---|
| bigbench_causal_judgement | 0 | multiple_choice_grade | 59.47 | ± | 3.57 |
| bigbench_date_understanding | 0 | multiple_choice_grade | 64.50 | ± | 2.49 |
| bigbench_disambiguation_qa | 0 | multiple_choice_grade | 44.96 | ± | 3.10 |
| bigbench_geometric_shapes | 0 | multiple_choice_grade | 22.84 | ± | 2.22 |
| exact_str_match | 2.79 | ± | 0.87 | ||
| bigbench_logical_deduction_five_objects | 0 | multiple_choice_grade | 30.80 | ± | 2.07 |
| bigbench_logical_deduction_seven_objects | 0 | multiple_choice_grade | 21.57 | ± | 1.56 |
| bigbench_logical_deduction_three_objects | 0 | multiple_choice_grade | 56.67 | ± | 2.87 |
| bigbench_movie_recommendation | 0 | multiple_choice_grade | 51.60 | ± | 2.24 |
| bigbench_navigate | 0 | multiple_choice_grade | 51.00 | ± | 1.58 |
| bigbench_reasoning_about_colored_objects | 0 | multiple_choice_grade | 70.35 | ± | 1.02 |
| bigbench_ruin_names | 0 | multiple_choice_grade | 51.79 | ± | 2.36 |
| bigbench_salient_translation_error_detection | 0 | multiple_choice_grade | 35.97 | ± | 1.52 |
| bigbench_snarks | 0 | multiple_choice_grade | 79.01 | ± | 3.04 |
| bigbench_sports_understanding | 0 | multiple_choice_grade | 75.66 | ± | 1.37 |
| bigbench_temporal_sequences | 0 | multiple_choice_grade | 47.90 | ± | 1.58 |
| bigbench_tracking_shuffled_objects_five_objects | 0 | multiple_choice_grade | 23.84 | ± | 1.21 |
| bigbench_tracking_shuffled_objects_seven_objects | 0 | multiple_choice_grade | 18.00 | ± | 0.92 |
| bigbench_tracking_shuffled_objects_three_objects | 0 | multiple_choice_grade | 56.67 | ± | 2.87 |