Skip to content

Commit

Permalink
added mock data
Browse files Browse the repository at this point in the history
  • Loading branch information
menamerai committed Feb 14, 2024
1 parent b8d0d8c commit 714e90b
Showing 1 changed file with 47 additions and 0 deletions.
47 changes: 47 additions & 0 deletions data/mock-per-token-performance.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,47 @@
from math import log2
from random import random, seed

seed(0)

performance_data = {
"llama2-100k": {
"nouns": [log2(random()) for _ in range(200)],
"verbs": [log2(random()) for _ in range(100)],
"adjectives": [log2(random()) for _ in range(250)],
"adverbs": [log2(random()) for _ in range(40)],
"pronouns": [log2(random()) for _ in range(55)],
"prepositions": [log2(random()) for _ in range(234)],
},
"llama2-1m": {
"nouns": [log2(random()) for _ in range(200)],
"verbs": [log2(random()) for _ in range(100)],
"adjectives": [log2(random()) for _ in range(250)],
"adverbs": [log2(random()) for _ in range(40)],
"pronouns": [log2(random()) for _ in range(55)],
"prepositions": [log2(random()) for _ in range(234)],
},
"llama2-10m": {
"nouns": [log2(random()) for _ in range(200)],
"verbs": [log2(random()) for _ in range(100)],
"adjectives": [log2(random()) for _ in range(250)],
"adverbs": [log2(random()) for _ in range(40)],
"pronouns": [log2(random()) for _ in range(55)],
"prepositions": [log2(random()) for _ in range(234)],
},
"llama2-100m": {
"nouns": [log2(random()) for _ in range(200)],
"verbs": [log2(random()) for _ in range(100)],
"adjectives": [log2(random()) for _ in range(250)],
"adverbs": [log2(random()) for _ in range(40)],
"pronouns": [log2(random()) for _ in range(55)],
"prepositions": [log2(random()) for _ in range(234)],
},
"llama2-1b": {
"nouns": [log2(random()) for _ in range(200)],
"verbs": [log2(random()) for _ in range(100)],
"adjectives": [log2(random()) for _ in range(250)],
"adverbs": [log2(random()) for _ in range(40)],
"pronouns": [log2(random()) for _ in range(55)],
"prepositions": [log2(random()) for _ in range(234)],
},
}

0 comments on commit 714e90b

Please sign in to comment.