tiny-shakespeare / README.md
aaronmac's picture
Upload folder using huggingface_hub
e6e1208 verified
|
Raw History Blame Contribute Delete
555 Bytes
---
library_name: transformers
pipeline_tag: text-generation
tags:
- causal-lm
- pytorch
- shakespeare
- educational
---
# Tiny Shakespeare Transformer
A small character-level causal Transformer trained from scratch on Tiny Shakespeare.
## Architecture
- Character tokenizer
- 4 transformer blocks
- 4 attention heads
- embedding dimension 64
- context length 32
## Loading
```python
from transformers import AutoModelForCausalLM
model = AutoModelForCausalLM.from_pretrained(
"YOUR_USERNAME/tiny-shakespeare",
trust_remote_code=True
)
```