Update README.md
Browse files
README.md
CHANGED
|
@@ -8,7 +8,21 @@ tags: []
|
|
| 8 |
<!-- Provide a quick summary of what the model is/does. -->
|
| 9 |
gpt2-medium + 16GTok from secret sauce mixture @ 2e-5 -> 2e-6 cosine schedule
|
| 10 |
|
| 11 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
## Model Details
|
| 13 |
|
| 14 |
### Model Description
|
|
|
|
| 8 |
<!-- Provide a quick summary of what the model is/does. -->
|
| 9 |
gpt2-medium + 16GTok from secret sauce mixture @ 2e-5 -> 2e-6 cosine schedule
|
| 10 |
|
| 11 |
+
```python
|
| 12 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 13 |
+
|
| 14 |
+
model = AutoModelForCausalLM.from_pretrained("crumb/gpt2-medium-eb49cc")
|
| 15 |
+
tokenizer = AutoTokenizer.from_pretrained("crumb/gpt2-medium-eb49cc")
|
| 16 |
+
|
| 17 |
+
prompt = "In a shocking finding, scientist discovered a herd of unicorns living in a remote, previously unexplored valley, in the Andes Mountains. Even more surprising to the researchers was the fact that the unicorns spoke perfect English."
|
| 18 |
+
input_ids = tokenizer(prompt, return_tensors="pt").input_ids
|
| 19 |
+
output = model.generate(input_ids, max_new_tokens=256, temperature=0.7, do_sample=True, penalty_alpha=0.6, top_k=16, repetition_penalty=1.1)
|
| 20 |
+
print(tokenizer.decode(output[0], skip_special_tokens=True))
|
| 21 |
+
# In a shocking finding, scientist discovered a herd of unicorns living in a remote, previously unexplored valley, in the Andes Mountains. Even more surprising to the researchers was the fact that the unicorns spoke perfect English.
|
| 22 |
+
# I'm not sure if they have any language barrier. What I do know is what makes such an amazing find possible and so compelling, and how it shows us something about the nature behind our shared universe!
|
| 23 |
+
# The researchers were studying the ecology on Mount Kilimanjaro when one day while hiking they stumbled across an unexpected sight: two unicorn herds grazing together. The scientists immediately realized there must be some sort - or possibly group-of organisms- that could help them understand this unusual animal behavior – especially considering its strange appearance…and the fact these animals can communicate with each other using sounds. It's as though they are communicating through the sound waves generated by their own horns. This discovery may prove crucial for understanding how we evolved to live in groups. For example...
|
| 24 |
+
# What does the unicorn mean? Well based off his name "Kilimander", this is a very interesting creature. He looks like he should come from another planet but has quite similar characteristics. His horn is covered up almost completely with skin so it's hard even touching him. So you would think he would use it only during mating season (when females want attention) although he might also use himself as prey during mating seasons. As a result male birds will often chase and mate directly below him to
|
| 25 |
+
```
|
| 26 |
## Model Details
|
| 27 |
|
| 28 |
### Model Description
|